Make use of enums in CastState even safer (skorokithakis#115)

* Make use of enums in CastState even safer * Add type info to CastState constructor * Add return type to type info for CastState constructor
syamlal · Sep 6, 2018 · e8a6a7d · e8a6a7d
1 parent f2275ab
commit e8a6a7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/catt/controllers.py b/catt/controllers.py
@@ -235,7 +235,7 @@ class StateMode(Enum):
 
 
 class CastState(CattStore):
-    def __init__(self, state_path, mode):
+    def __init__(self, state_path: Path, mode: StateMode) -> None:
         super(CastState, self).__init__(state_path)
         if mode == StateMode.CONF:
             self._create_store_dir()