fix NotImplementedError string

nod-ai · Jun 14, 2023 · e951028 · e951028
1 parent 9f3f9b0
commit e951028
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/test_tensor.py b/tests/unit/test_tensor.py
@@ -55,7 +55,7 @@ def test_non_infer_returns(self):
             except NotImplementedError as e:
                 assert (
                     str(e)
-                    == "__mod__ with signature __mod__(self, other Any) -> Tensor"
+                    == "NotImplementedError: __mod__ with signature __mod__(self, other Any) -> Tensor"
                 )
 
     def test_optional_args(self):