Update

[ghstack-poisoned]
pytorch · guilhermeleobas · Apr 7, 2025 · Apr 7, 2025 · Apr 8, 2025 · Apr 8, 2025
commit c4ee847bedc0cd41b7a549c99db83e84217ae414
diff --git a/torch/_dynamo/test_case.py b/torch/_dynamo/test_case.py
@@ -15,6 +15,7 @@
 import inspect
 import logging
 import os
+import pathlib
 import re
 import sys
 import unittest
@@ -174,20 +175,18 @@ def tearDownClass(cls) -> None:
     @classmethod
     def setUpClass(cls) -> None:
         # Skip test if python versions doesn't match
-        m = re.search(r"\b\d+_\d+\b", inspect.getfile(cls))
+        normalized_path = pathlib.PurePath("dynamo/cpython").as_posix()
+        regex = re.escape(normalized_path) + r"\b\d+_\d{2}\b"
+        m = re.search(regex, inspect.getfile(cls))
         if m:
             test_py_ver = tuple(map(int, m.group().split("_")))
-        else:
-            raise unittest.TestCase.failureException(
-                f"Test file {inspect.getfile(cls)} does not contain a valid Python version"
-            )
-        py_ver = sys.version_info[:2]
-        if py_ver != test_py_ver:
-            expected = ".".join(map(str, test_py_ver))
-            got = ".".join(map(str, py_ver))
-            raise unittest.SkipTest(
-                f"Test requires Python {expected} but got Python {got}"
-            )
+            py_ver = sys.version_info[:2]
+            if py_ver != test_py_ver:
+                expected = ".".join(map(str, test_py_ver))
+                got = ".".join(map(str, py_ver))
+                raise unittest.SkipTest(
+                    f"Test requires Python {expected} but got Python {got}"
+                )
 
         super().setUpClass()
         cls._stack = contextlib.ExitStack()  # type: ignore[attr-defined]