Assign encoder and model tasks to same worker

gabegma · gabegma · commit 853dae958be1 · 2023-04-19T14:25:40.000-04:00
diff --git a/azimuth/modules/base_classes/dask_module.py b/azimuth/modules/base_classes/dask_module.py
@@ -28,7 +28,7 @@
 
 class Worker(IntEnum):
     model = 0
-    encoder = 1
+    encoder = 0
 
 
 class DaskModule(HDF5CacheMixin, Generic[ConfigScope]):
