Update setup.py

itskalvik · Aug 17, 2024 · 42827ed · 42827ed
1 parent 409d0d9
commit 42827ed
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -24,9 +24,8 @@
                       'cma',
                       'bayesian-optimization',
                       'hkb_diamondsquare',
-                      'tensorflow-probability[tf]>=0.21.0',
-                      'tensorflow>=2.13.0; platform_machine!="arm64"',
-                      'tensorflow-aarch64>=2.13.0; platform_machine=="arm64"',
+                      'tensorflow-probability>=0.21.0',
+                      'tensorflow>=2.13.0',
                       'typing_extensions',
                       'gpflow>=2.7.0'
                      ]