Raise NotImplementedError from pytorch_typify

Author: HarshvirSandhu

pytensor/link/pytorch/dispatch/basic.py

@@ -16,7 +16,9 @@ def pytorch_typify(data, dtype=None, **kwargs):
     try:
         return torch.as_tensor(data, dtype=dtype)
     except RuntimeError:
-        raise RuntimeError(f"Data is of type {type(data)}, it should be an array")
+        raise NotImplementedError(
+            f"pytorch_typify got type {type(data)}, it should be an array"
+        )
 
 
 @pytorch_typify.register(NoneType)

tests/link/pytorch/test_subtensor.py

@@ -51,7 +51,7 @@ def test_pytorch_Subtensor():
     out_fg = FunctionGraph([x_pt], [out_pt])
     compare_pytorch_and_py(out_fg, [x_np])
 
-    with pytest.raises(RuntimeError):
+    with pytest.raises(NotImplementedError):
         out_pt = x_pt[[1, 2], :, [3, 4]]
         assert isinstance(out_pt.owner.op, pt_subtensor.AdvancedSubtensor)
         out_fg = FunctionGraph([x_pt], [out_pt])