diff --git a/conda.yml b/conda.yml
index 220ec0b90..57b5aead9 100644
--- a/conda.yml
+++ b/conda.yml
@@ -3,9 +3,12 @@ channels:
   - conda-forge
   - defaults
 dependencies:
+  - python=3.8.17
   - mlflow=1.14.1
   - pyyaml=5.3.1
   - hydra-core=1.0.6
   - pip=20.3.3
+  - numpy=1.19.5
   - pip:
-      - wandb==0.10.31
\ No newline at end of file
+      - wandb==0.10.31
+      - protobuf==3.20.0
\ No newline at end of file
diff --git a/environment.yml b/environment.yml
index efabac598..94866ae8b 100644
--- a/environment.yml
+++ b/environment.yml
@@ -3,6 +3,7 @@ channels:
   - conda-forge
   - defaults
 dependencies:
+  - python=3.8.17
   - mlflow=1.14.1
   - ipython=7.21.0
   - notebook=6.2.0
@@ -13,5 +14,7 @@ dependencies:
   - pandas=1.2.3
   - git=2.30.2
   - pip=20.3.3
+  - numpy=1.19.5
   - pip:
       - wandb==0.10.31
+      - protobuf==3.20.0
\ No newline at end of file