fix Eq on TValue

sonos · Sep 5, 2023 · 889f82e · 889f82e
1 parent c6a8d78
commit 889f82e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/core/src/value.rs b/core/src/value.rs
@@ -1,10 +1,11 @@
 use crate::internal::*;
+use std::ops::Deref;
 use std::rc::Rc;
 
 use tract_ndarray::Array;
 use TValue::*;
 
-#[derive(Clone, PartialEq, Eq)]
+#[derive(Clone, Eq)]
 pub enum TValue {
     Const(Arc<Tensor>),
     Var(Rc<Tensor>),
@@ -16,6 +17,12 @@ impl std::fmt::Debug for TValue {
     }
 }
 
+impl PartialEq for TValue {
+    fn eq(&self, other: &Self) -> bool {
+        self.deref() == other.deref()
+    }
+}
+
 impl TValue {
     pub fn is_exclusive(&self) -> bool {
         match self {