Small fix

project-lighter · May 29, 2024 · 3b37a20 · 3b37a20
1 parent f4d4dab
commit 3b37a20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/test_cifar.py b/tests/integration/test_cifar.py
@@ -22,5 +22,5 @@ def test_trainer_method(method_name: str, config_file: str):
     """ """
     kwargs = {"config_file": config_file, "args_file": test_overrides}
 
-    func_return = run_trainer_method(method_name, **kwargs)
+    func_return = run(method_name, **kwargs)
     assert func_return is None