lintrunner

pytorch · Feb 9, 2023 · 98a5bcd · 98a5bcd
1 parent ac5f5d9
commit 98a5bcd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torch/_prims/nvfuser_prims.py b/torch/_prims/nvfuser_prims.py
@@ -144,7 +144,9 @@
 def _assert_nvfuser_op_exists(fname: str):
     try:
         try:
-            from nvfuser import FusionDefinition as fd  # type: ignore[import, attr-defined]
+            from nvfuser import (  # type: ignore[import, attr-defined]
+                FusionDefinition as fd,
+            )
         except ImportError:
             from nvfuser._C import FusionDefinition as fd  # type: ignore[import]