[fix] temporarily adding Any type

uhlmanngroup · Aug 21, 2024 · 728b3f7 · 728b3f7
1 parent 092800a
commit 728b3f7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bioimage_embed/config.py b/bioimage_embed/config.py
@@ -185,7 +185,8 @@ class Trainer:
     # logger: Optional[any]
     gradient_clip_val: float = 0.5
     enable_checkpointing: bool = True
-    devices: str = "auto"
+    devices: Any = "auto"
+    num_nodes: int = 1
     accelerator: str = "auto"
     accumulate_grad_batches: int = 16
     min_epochs: int = 1