diff --git a/environment.yml b/environment.yml
index e90d50aa167323961427c1e70e8bd5ada2680d9d..23388554ab9be10fab8cdcabf8c904f6691b01a7 100644
--- a/environment.yml
+++ b/environment.yml
@@ -6,3 +6,4 @@ dependencies:
   - matplotlib
   - scikit-learn
   - time
+  - tensorflow
\ No newline at end of file