diff --git a/nemo_run/run/experiment.py b/nemo_run/run/experiment.py index ba430413..c0b93a8a 100644 --- a/nemo_run/run/experiment.py +++ b/nemo_run/run/experiment.py @@ -291,7 +291,7 @@ def from_title( parent_dir = os.path.join(get_nemorun_home(), "experiments", title) exp_dir = _get_latest_dir(parent_dir) - assert os.path.isdir(exp_dir), f"Experiment {id} not found." + assert os.path.isdir(exp_dir), f"Experiment {title} not found." exp = cls._from_config(exp_dir) return exp diff --git a/test/run/test_experiment.py b/test/run/test_experiment.py index 4f160237..d750dca9 100644 --- a/test/run/test_experiment.py +++ b/test/run/test_experiment.py @@ -436,7 +436,7 @@ def test_from_title_nonexistent(temp_dir): mock_get_latest_dir.return_value = nonexistent_path # The assertion should fail because the directory doesn't exist - with pytest.raises(AssertionError): + with pytest.raises(AssertionError, match=f"Experiment {title} not found"): Experiment.from_title(title)