fix: 🐛 Fix LocalTorch distribute_workers

janelia-cellmap · May 29, 2024 · 1b90080 · 1b90080
1 parent a3a8f1e
commit 1b90080
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/dacapo/compute_context/compute_context.py b/dacapo/compute_context/compute_context.py
@@ -7,12 +7,6 @@
 
 
 class ComputeContext(ABC):
-    distribute_workers: Optional[bool] = attr.ib(
-        default=False,
-        metadata={
-            "help_text": "Whether to distribute the workers across multiple nodes or processes."
-        },
-    )
     """
     The ComputeContext class is an abstract base class for defining the context in which computations are to be done.
     It is inherited from the built-in class `ABC` (Abstract Base Classes). Other classes can inherit this class to define

diff --git a/dacapo/compute_context/local_torch.py b/dacapo/compute_context/local_torch.py
@@ -23,6 +23,12 @@ class LocalTorch(ComputeContext):
         The class is a subclass of the ComputeContext class.
     """
 
+    distribute_workers: Optional[bool] = attr.ib(
+        default=False,
+        metadata={
+            "help_text": "Whether to distribute the workers across multiple nodes or processes."
+        },
+    )
     _device: Optional[str] = attr.ib(
         default=None,
         metadata={