chore: fix up bad requirements

ZettaAI · Dec 1, 2023 · a9fd31f · a9fd31f
1 parent 33047da
commit a9fd31f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -96,6 +96,8 @@ training = [
     "pytorch-lightning ~= 2.0.0",  # mp dataloader causes segmentation faults with 2.1.0
     "torchmetrics == 0.11.4",
     "wandb >= 0.13.1",
+    "kubernetes",
+    "zetta_utils[mazepa]",
 ]
 datastore = ["google-cloud-datastore"]
 docs = [