Cleanup requirements (#605)

anakinxc · web-flow · commit 667170275c1b · 2024-03-12T21:36:58.000+08:00
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -14,3 +14,6 @@ linkify-it-py==2.0.3
 mdutils==1.6.0
 spu>=0.3.1b0
 pydata-sphinx-theme
+tabulate
+xlwt
+xlsxwriter
diff --git a/examples/python/ml/requirements.txt b/examples/python/ml/requirements.txt
@@ -6,3 +6,6 @@ torch==2.2.0
 torch_xla==2.2.0
 torchvision
 jax[cpu]<=0.4.23
+tensorflow>=2.12.0
+tensorflow_datasets
+keras
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -1,13 +1,5 @@
-matplotlib
-tensorflow >= 2.12.0; platform_machine != "arm64"
-tensorflow_datasets
 pandas>=1.4.2
 flax
-keras
 scikit-learn
 # for tests
 absl-py>=1.1.0
-# for benchmark
-tabulate
-xlwt
-xlsxwriter
