Add missing AVAIL_MACHINES

lhoyer · May 28, 2021 · 269cc3a · 269cc3a
1 parent 76c1fd0
commit 269cc3a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/configs/machine_config.py b/configs/machine_config.py
@@ -3,6 +3,7 @@ class MachineConfig:
     CITYSCAPES_DIR = None
     GENERATED_DEPTH_DIR = None
     LOG_DIR = None
+    AVAIL_MACHINES = ["ws"]
 
     def __init__(self, machine):
         if machine == "ws":