Update test_model.py

Fix typo in CUDA test.
torchmd · Mar 19, 2024 · 9546e88 · 9546e88
1 parent fae79bd
commit 9546e88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -119,7 +119,7 @@ def test_cuda_graph_compatible(model_name):
             "prior_model": None,
             "atom_filter": -1,
             "derivative": True,
-            "check_error": False,
+            "check_errors": False,
             "static_shapes": True,
             "output_model": "Scalar",
             "reduce_op": "sum",