Type annotation

NVIDIA · Jan 19, 2025 · d7dc6cd · d7dc6cd
1 parent 39efb76
commit d7dc6cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nvfuser/__init__.py b/nvfuser/__init__.py
@@ -372,7 +372,8 @@ def execute(
                 _enable_options=_enable_options,
                 _disable_options=_disable_options,
             )
-            out_tensors = []
+
+            out_tensors: list[torch.Tensor] = []
             for out_dtensor in out_dtensors:
                 if out_dtensor.mesh.size == 0:
                     out_tensors.append(out_dtensor.local)