diff --git a/tests/unit/test_job_to_slurm.py b/tests/unit/test_job_to_slurm.py index c11cde5..936a0e7 100644 --- a/tests/unit/test_job_to_slurm.py +++ b/tests/unit/test_job_to_slurm.py @@ -10,9 +10,7 @@ def mock_load_config(tmp_path): mock_config = {"local": {"path": tmp_path}} os.makedirs(tmp_path / "jobs", exist_ok=True) - with patch( - "milex_scheduler.load_config", return_value=mock_config - ) as mock_load_config: + with patch("load_config", return_value=mock_config) as mock_load_config: yield mock_load_config