Merge branch 'main' into wjy/dist

NVIDIA · Jan 20, 2025 · ed9fbbf · ed9fbbf
2 parents d7dc6cd + aa3c3d3
commit ed9fbbf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/nvfuser/__init__.py b/nvfuser/__init__.py
@@ -6,7 +6,7 @@
 import os
 import re
 import sys
-from typing import Callable, Optional, Union, List, Iterable  # noqa: F401
+from typing import Callable, Iterable
 import warnings
 
 import torch
@@ -610,8 +610,8 @@ def _repro_error_str(self, section: str, inputs: list | None = None):
 
     def validate(
         self,
-        inputs: List[torch.Tensor],
-        reference_outputs: List[torch.Tensor],
+        inputs: list[torch.Tensor],
+        reference_outputs: list[torch.Tensor],
         kwargs=None,
     ):
         """

diff --git a/tests/python/utils.py b/tests/python/utils.py
@@ -288,7 +288,7 @@ def check_cpp_translation(
             "(A failure here suggests a mismatch in functionality between the original and cloned definitions.)"
         )
         print("Does FusionDefinition supports segmentation?\t", supports_segmentation)
-        print(fd.getReproErrorString("executing", inputs))
+        print(fd._repro_error_str("executing", inputs))
         raise err