Commit 954967b8 authored by Jonas Kaufmann's avatar Jonas Kaufmann Committed by Antoine Kaufmann
Browse files

orchestration/ExpEnv: add type annotations for constructor

parent 6b3fc27f
...@@ -30,7 +30,7 @@ if tp.TYPE_CHECKING: # prevent cyclic import ...@@ -30,7 +30,7 @@ if tp.TYPE_CHECKING: # prevent cyclic import
class ExpEnv(object): class ExpEnv(object):
"""Manages the experiment environment.""" """Manages the experiment environment."""
def __init__(self, repo_path, workdir, cpdir) -> None: def __init__(self, repo_path: str, workdir: str, cpdir: str) -> None:
self.create_cp = False self.create_cp = False
"""Whether a checkpoint should be created.""" """Whether a checkpoint should be created."""
self.restore_cp = False self.restore_cp = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment