fix pytest deprecations

geoopt · Aug 4, 2020 · e390bc9 · e390bc9
1 parent 5dd153f
commit e390bc9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_gyrovector_math.py b/tests/test_gyrovector_math.py
@@ -10,7 +10,7 @@
 from geoopt.manifolds import stereographic
 
 
-@pytest.fixture("function", autouse=True, params=range(30, 40))
+@pytest.fixture(scope="function", autouse=True, params=range(30, 40))
 def seed(request):
     seed = request.param
     torch.manual_seed(seed)
@@ -20,7 +20,7 @@ def seed(request):
 
 
 @pytest.fixture(
-    "function", params=[torch.float64, torch.float32], ids=["float64", "float32"]
+    scope="function", params=[torch.float64, torch.float32], ids=["float64", "float32"]
 )
 def dtype(request):
     return request.param

diff --git a/tests/test_lorentz_math.py b/tests/test_lorentz_math.py
@@ -5,7 +5,7 @@
 from geoopt.manifolds import lorentz
 
 
-@pytest.fixture("module", autouse=True, params=range(30, 40))
+@pytest.fixture(scope="module", autouse=True, params=range(30, 40))
 def seed(request):
     seed = request.param
     torch.manual_seed(seed)
@@ -14,7 +14,7 @@ def seed(request):
     return seed
 
 
-@pytest.fixture("module", params=[torch.float64, torch.float32])
+@pytest.fixture(scope="module", params=[torch.float64, torch.float32])
 def dtype(request):
     return request.param