Move jinja2 check lower

spcl · Oct 30, 2024 · 0c8cd0b · 0c8cd0b
1 parent 9267904
commit 0c8cd0b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/dace/cli/sdfv.py b/dace/cli/sdfv.py
@@ -34,11 +34,6 @@ def view(sdfg: dace.SDFG, filename: Optional[Union[str, int]] = None, verbose: b
                     blocking the current thread.
     :param verbose: Be verbose.
     """
-    try:
-        import jinja2
-    except (ImportError, ModuleNotFoundError):
-        raise ImportError('SDFG.view() requires jinja2, please install by running `pip install jinja2`')
-
     # If vscode is open, try to open it inside vscode
     if filename is None:
         if (
@@ -52,6 +47,11 @@ def view(sdfg: dace.SDFG, filename: Optional[Union[str, int]] = None, verbose: b
             os.close(fd)
             return
 
+    try:
+        import jinja2
+    except (ImportError, ModuleNotFoundError):
+        raise ImportError('SDFG.view() requires jinja2, please install by running `pip install jinja2`')
+
     if type(sdfg) is dace.SDFG:
         sdfg = dace.serialize.dumps(sdfg.to_json())