[Minor] Remove warnings in test_cost (#1902)

pytorch · Feb 12, 2024 · 2f9e1ae · 2f9e1ae
1 parent 69d44f5
commit 2f9e1ae
Show file tree

Hide file tree

Showing 6 changed files with 165 additions and 43 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -53,7 +53,7 @@ def fin():
     request.addfinalizer(fin)
 
 
-@pytest.fixture(scope="session", autouse=True)
+@pytest.fixture(autouse=True)
 def set_warnings() -> None:
     warnings.filterwarnings(
         "ignore",
@@ -65,6 +65,11 @@ def set_warnings() -> None:
         category=UserWarning,
         message=r"Couldn't cast the policy onto the desired device on remote process",
     )
+    warnings.filterwarnings(
+        "ignore",
+        category=UserWarning,
+        message=r"Skipping device Apple Paravirtual device",
+    )
     warnings.filterwarnings(
         "ignore",
         category=DeprecationWarning,