Remove custom equality comparison

rapidsai · Nov 18, 2024 · 7c8a804 · 7c8a804
1 parent 4c0823d
commit 7c8a804
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/legateboost/models/krr.py b/legateboost/models/krr.py
@@ -221,11 +221,3 @@ def __mul__(self, scalar: Any) -> "KRR":
         new = copy.deepcopy(self)
         self.betas_ *= scalar
         return new
-
-    def __eq__(self, other: object) -> bool:
-        if not isinstance(other, KRR):
-            raise NotImplementedError()
-        return bool(
-            (other.betas_ == self.betas_).all()
-            and (other.X_train == self.X_train).all()
-        )
diff --git a/legateboost/models/linear.py b/legateboost/models/linear.py
@@ -123,8 +123,3 @@ def __mul__(self, scalar: Any) -> "Linear":
         new = copy.deepcopy(self)
         new.betas_ *= scalar
         return new
-
-    def __eq__(self, other: object) -> bool:
-        if not isinstance(other, Linear):
-            raise NotImplementedError()
-        return bool((other.betas_ == self.betas_).all())