Remove unused import, use torch mode

Author: Ch0ronomato

pytensor/link/pytorch/dispatch/__init__.py

@@ -11,7 +11,6 @@
 import pytensor.link.pytorch.dispatch.shape
 import pytensor.link.pytorch.dispatch.sort
 import pytensor.link.pytorch.dispatch.subtensor
-import pytensor.link.pytorch.dispatch.nlinalg
 import pytensor.link.pytorch.dispatch.slinalg
 import pytensor.link.pytorch.dispatch.blockwise
 # isort: on

tests/link/pytorch/test_slinalg.py

@@ -6,6 +6,9 @@
 from pytensor.tensor.slinalg import cholesky
 
 
+torch = pytest.importorskip("torch")
+
+
 @pytest.mark.parametrize(
     "cov_batch_shape", [(), (1000,), (4, 1000)], ids=lambda arg: f"cov:{arg}"
 )
@@ -19,4 +22,6 @@ def test_batched_mvnormal_logp_and_dlogp(cov_batch_shape):
     chol_cov = cholesky(cov, lower=True, on_error="raise")
 
     fn = pytensor.function([cov], [chol_cov])
-    assert np.all(np.isclose(fn(test_values), np.linalg.cholesky(test_values)))
+    assert np.all(
+        np.isclose(fn(test_values, mode="PYTORCH"), np.linalg.cholesky(test_values))
+    )