Fix imports

Author: Ian Schweer

tests/link/pytorch/test_basic.py

@@ -4,7 +4,6 @@
 import numpy as np
 import pytest
 
-import pytensor.tensor as pt
 import pytensor.tensor.basic as ptb
 from pytensor.compile.builders import OpFromGraph
 from pytensor.compile.function import function
@@ -18,7 +17,7 @@
 from pytensor.ifelse import ifelse
 from pytensor.link.pytorch.linker import PytorchLinker
 from pytensor.raise_op import CheckAndRaise
-from pytensor.tensor import alloc, arange, as_tensor, empty, eye
+from pytensor.tensor import alloc, arange, as_tensor, empty, expit, eye, softplus
 from pytensor.tensor.type import matrices, matrix, scalar, vector
 
 
@@ -348,13 +347,13 @@ def test_pytorch_OpFromGraph():
 
 def test_pytorch_scipy():
     x = vector("a", shape=(3,))
-    out = pt.expit(x)
+    out = expit(x)
     f = FunctionGraph([x], [out])
     compare_pytorch_and_py(f, [np.random.rand(3)])
 
 
 def test_pytorch_softplus():
     x = vector("a", shape=(3,))
-    out = pt.softplus(x)
+    out = softplus(x)
     f = FunctionGraph([x], [out])
     compare_pytorch_and_py(f, [np.random.rand(3)])