Update conftest.py

ivy-llc · Aug 16, 2023 · 8ab26fc · 8ab26fc
1 parent fedc3d0
commit 8ab26fc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/conftest.py b/conftest.py
@@ -12,7 +12,7 @@
 
 
 @pytest.fixture(autouse=True)
-def run_around_tests(dev_str, compile_graph, fw):
+def run_around_tests(dev_str, f, compile_graph, fw):
     if "gpu" in device and fw == "numpy":
         # Numpy does not support GPU
         pytest.skip()
@@ -50,8 +50,8 @@ def pytest_generate_tests(metafunc):
     for backend_str in backend_strs:
         for device in devices:
             for compile_graph in compile_modes:
-                configs.append((device, compile_graph, backend_str))
-    metafunc.parametrize("dev_str,compile_graph,fw", configs)
+                configs.append((device, None, compile_graph, backend_str))
+    metafunc.parametrize("dev_str,f,compile_graph,fw", configs)
 
 
 def pytest_addoption(parser):