tests: Add 'pytorch' mark to all 'PyTorch' execution mode variants (#…

…2774) Tests that need pytorch in other execution modes (e.g. for construction) still need to be marked explicitly. Signed-off-by: Jan Vesely <[email protected]>
PrincetonUniversity · Aug 17, 2023 · d2a72b9 · d2a72b9
1 parent 233a0ce
commit d2a72b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conftest.py b/conftest.py
@@ -81,7 +81,7 @@ def pytest_generate_tests(metafunc):
     if "autodiff_mode" in metafunc.fixturenames:
         auto_modes = [
             # pnlvm.ExecutionMode.Python,
-            pnlvm.ExecutionMode.PyTorch,
+            pytest.param(pnlvm.ExecutionMode.PyTorch, marks=pytest.mark.pytorch),
             pytest.param(pnlvm.ExecutionMode.LLVMRun, marks=pytest.mark.llvm)
         ]
         metafunc.parametrize("autodiff_mode", auto_modes)