Removed temporary fixes

pytorch · Nov 7, 2023 · 7f91053 · 7f91053
1 parent 829a2d3
commit 7f91053
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py
@@ -221,13 +221,7 @@ def check_functional(functional, input, *args, check_scripted_smoke=True, check_
 
     # Skip check on Windows as torch.compile does not work on Win32
     if check_torch_compile_smoke and sys.platform != "win32":
-        # Temporary fix to catch deprectation warning
-        # This can be removed once https://github.com/pytorch/pytorch/pull/113023 is merged:
-        import warnings
-
-        with warnings.catch_warnings():
-            warnings.filterwarnings("ignore", category=DeprecationWarning)
-            _check_functional_torch_compile_smoke(functional, input, *args, **kwargs)
+        _check_functional_torch_compile_smoke(functional, input, *args, **kwargs)
 
 
 def check_functional_kernel_signature_match(functional, *, kernel, input_type):
@@ -4137,7 +4131,7 @@ def test_kernel_video(self):
 
     @pytest.mark.parametrize("make_input", [make_image_tensor, make_image_pil, make_image, make_video])
     def test_functional(self, make_input):
-        check_functional(F.equalize, make_input(), check_torch_compile_smoke=False)
+        check_functional(F.equalize, make_input())
 
     @pytest.mark.parametrize(
         ("kernel", "input_type"),