diff --git a/environment.yml b/environment.yml
index b49f482c..c45573a0 100644
--- a/environment.yml
+++ b/environment.yml
@@ -22,3 +22,4 @@ dependencies:
       - pysr==0.16.3
       - xgboost==1.7.5
       - zstd
+      - tensorflow-model-optimization