Late import torch

Author: Ian Schweer

pytensor/link/pytorch/linker.py

@@ -1,8 +1,6 @@
 import copy
 from typing import Any
 
-from torch import is_tensor
-
 from pytensor.graph.basic import Variable
 from pytensor.link.basic import JITLinker
 from pytensor.link.utils import unique_name_generator
@@ -21,6 +19,8 @@ def input_filter(self, inp: Any) -> Any:
         return pytorch_typify(inp)
 
     def output_filter(self, var: Variable, out: Any) -> Any:
+        from torch import is_tensor
+
         if is_tensor(out):
             return out.cpu()
         else: