Fix typehints

This commit is contained in:
Marek Wolan
2023-11-16 15:28:38 +00:00
parent 13c49bf3ea
commit 0b9bdedebd
5 changed files with 8 additions and 10 deletions

View File

@@ -40,8 +40,6 @@ 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)