From a9fd31f8d7e5d760f776a8e1e1d129ba3af34648 Mon Sep 17 00:00:00 2001
From: supersergiy <sergiy.popovich@gmail.com>
Date: Fri, 1 Dec 2023 15:40:06 -0800
Subject: [PATCH] chore: fix up bad requirements

---
 pyproject.toml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index d87eece91..4b1fdc6f0 100644
--- 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 = [