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
class ExpEnv(object):
"""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
"""Whether a checkpoint should be created."""
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