diff --git a/tests/conftest.py b/tests/conftest.py index fe450213..6a65b12f 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -100,7 +100,6 @@ class TempPrimaiteSession(PrimaiteSession): @pytest.fixture def temp_primaite_session(request, monkeypatch) -> TempPrimaiteSession: """Create a temporary PrimaiteSession object.""" - with monkeypatch.context() as m: - m.setattr(PRIMAITE_PATHS, "user_sessions_path", temp_user_sessions_path()) - config_path = request.param[0] - return TempPrimaiteSession.from_config(config_path=config_path) + monkeypatch.setattr(PRIMAITE_PATHS, "user_sessions_path", temp_user_sessions_path()) + config_path = request.param[0] + return TempPrimaiteSession.from_config(config_path=config_path) diff --git a/tests/e2e_integration_tests/test_primaite_session.py b/tests/e2e_integration_tests/test_primaite_session.py index 5e1da4ff..c6179e9a 100644 --- a/tests/e2e_integration_tests/test_primaite_session.py +++ b/tests/e2e_integration_tests/test_primaite_session.py @@ -40,6 +40,8 @@ class TestPrimaiteSession: with temp_primaite_session as session: session: TempPrimaiteSession session.start_session() + for i in range(100): + print(session.io_manager.generate_session_path()) # TODO: include checks that the model was trained, e.g. that the loss changed and checkpoints were saved? @pytest.mark.parametrize("temp_primaite_session", [[EVAL_ONLY_PATH]], indirect=True)