Tensor eq

rwth-i6 · Mar 16, 2023 · e0541e4 · e0541e4
1 parent d69f279
commit e0541e4
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/returnn/tensor/_tensor_op_overloads.py b/returnn/tensor/_tensor_op_overloads.py
@@ -16,13 +16,8 @@ class _TensorOpOverloadsMixin(_TensorMixinBase):
 
     # --- comparisons
 
-    # _TensorMixin.__eq__ is disabled as per the following error in some TF tests:
-    # AssertionError: unhashable type: 'Tensor'.
-    # See CI https://github.com/rwth-i6/returnn/actions/runs/4406240591
-    """
     def __eq__(self: Tensor, other: Union[_frontend_api.RawTensorTypes, Tensor]) -> Tensor:
         return self.raw_frontend.compare(self, "==", other)
-    """
 
     def __ne__(self: Tensor, other: Union[_frontend_api.RawTensorTypes, Tensor]) -> Tensor:
         return self.raw_frontend.compare(self, "!=", other)