make pytato a soft dependency again

Author: alexfikl

arraycontext/impl/pytato/__init__.py

@@ -219,18 +219,18 @@ def get_target(self):
 
 # {{{ PytatoPyOpenCLArrayContext
 
-from pytato.target.loopy import LoopyPyOpenCLTarget
-
-
-class _ArgSizeLimitingPytatoLoopyPyOpenCLTarget(LoopyPyOpenCLTarget):
-    def __init__(self, limit_arg_size_nbytes: int) -> None:
-        super().__init__()
-        self.limit_arg_size_nbytes = limit_arg_size_nbytes
-
-    @memoize_method
-    def get_loopy_target(self) -> Optional["lp.PyOpenCLTarget"]:
-        from loopy import PyOpenCLTarget
-        return PyOpenCLTarget(limit_arg_size_nbytes=self.limit_arg_size_nbytes)
+if "pytato" in sys.modules:
+    from pytato.target.loopy import LoopyPyOpenCLTarget
+
+    class _ArgSizeLimitingPytatoLoopyPyOpenCLTarget(LoopyPyOpenCLTarget):
+        def __init__(self, limit_arg_size_nbytes: int) -> None:
+            super().__init__()
+            self.limit_arg_size_nbytes = limit_arg_size_nbytes
+
+        @memoize_method
+        def get_loopy_target(self) -> Optional["lp.PyOpenCLTarget"]:
+            from loopy import PyOpenCLTarget
+            return PyOpenCLTarget(limit_arg_size_nbytes=self.limit_arg_size_nbytes)
 
 
 class PytatoPyOpenCLArrayContext(_BasePytatoArrayContext):