refactor requirements

Author: Sam-Armstrong

requirements/optional.txt

@@ -1,26 +1,28 @@
-h5py # unpinned
-pytest # unpinned
-networkx # unpinned
-hypothesis==6.98.10 # fixed version till this get solved: https://github.com/HypothesisWorks/hypothesis/issues/3896
-pymongo # unpinned
-redis # unpinned
-matplotlib # unpinned
-opencv-python # unpinned , mod_name=cv2
-jax[cpu] # unpinned, we test for latest version now, mod_name=jax
-jaxlib # unpinned, we test for latest version now
-paddlepaddle # unpinned , mod_name=paddle
-tensorflow # unpinned
-torch # unpinned, we test for latest version now
-scipy # unpinned
-dm-haiku # unpinned mod_name=haiku
+# frameworks
+dm-haiku # mod_name=haiku
 flax
-pydriller
-coverage
-scikit-learn # mod_name=sklearn
+jax[cpu]
+jaxlib
+paddlepaddle # mod_name=paddle
 pandas
-pyspark
+scikit-learn # mod_name=sklearn
+scipy
+tensorflow
+torch
+torchvision
+xgboost
+
+# other developer requirements
 autoflake # for backend generation
+coverage
+h5py
+hypothesis==6.98.10 # fixed version till this get solved: https://github.com/HypothesisWorks/hypothesis/issues/3896
+matplotlib
+networkx
+opencv-python # mod_name=cv2
+pydriller
+pymongo
+pyspark
+pytest
+redis
 snakeviz # for profiling
-cryptography
-xgboost
-torchvision

requirements/optional_apple_silicon_1.txt

@@ -1,16 +1,15 @@
 h5py
-pytest
-networkx
 hypothesis
-pymongo
-redis
-matplotlib
-opencv-python  # mod_name=cv2
 jax
 jaxlib
+matplotlib
+networkx
+opencv-python  # mod_name=cv2
+paddlepaddle  # unpinned , mod_name=paddle
+pymongo
+pytest
+redis
 tensorflow-macos  # mod_name=tensorflow_macos
 tensorflow-probability  # mod_name=tensorflow_probability
 torch
-paddlepaddle  # unpinned , mod_name=paddle
 torchvision
-cryptography

requirements/optional_apple_silicon_2.txt

@@ -1,12 +1,12 @@
-scipy
+autoflake  # for backend generation
+coverage
 dm-haiku  # mod_name=haiku
 flax
+ml-dtypes==0.2.0
+pandas
 protobuf
 pydriller
-coverage
-scikit-learn  # mod_name=sklearn
-pandas
 pyspark
-autoflake  # for backend generation
+scikit-learn  # mod_name=sklearn
+scipy
 snakeviz  # for profiling
-ml-dtypes==0.2.0

requirements/optional_apple_silicon_gpu_1.txt

@@ -1,17 +1,17 @@
 h5py
-pytest
-networkx
-hypothesis
-pymongo
-redis
-matplotlib
-opencv-python  # mod_name=cv2
 jax
 jaxlib
 jax-metal
+hypothesis
+matplotlib
+networkx
+opencv-python  # mod_name=cv2
+paddlepaddle  # unpinned , mod_name=paddle
+pymongo
+pytest
+redis
 tensorflow-macos  # mod_name=tensorflow_macos
 tensorflow-metal
 tensorflow-probability  # mod_name=tensorflow_probability
 torch
-paddlepaddle  # unpinned , mod_name=paddle
 torchvision

requirements/optional_apple_silicon_gpu_2.txt

@@ -1,12 +1,12 @@
-scipy
+autoflake  # for backend generation
+coverage
 dm-haiku  # mod_name=haiku
 flax
+ml-dtypes==0.2.0
+pandas
 protobuf
 pydriller
-coverage
-scikit-learn  # mod_name=sklearn
-pandas
 pyspark
-autoflake  # for backend generation
+scikit-learn  # mod_name=sklearn
+scipy
 snakeviz  # for profiling
-ml-dtypes==0.2.0

requirements/optional_applied.txt

@@ -1,19 +1,19 @@
-kornia
-torch-geometric
+accelerate
 botorch
+datasets
+diffusers
+fastai
+flax
+flower
 gpytorch
 gradslam
-fastai
+kornia
 pytorch-lightning
+rllib
+seaborn
 tensorly
 thinc
-transformers
-accelerate
-diffusers
 timm
-flax
-flower
-rllib
-seaborn
-datasets
+torch-geometric
+transformers
 validators

requirements/optional_gpu.txt

@@ -1,23 +1,23 @@
-h5py # unpinned
-pytest # unpinned
-networkx # unpinned
-hypothesis # unpinned
-pymongo # unpinned
-redis # unpinned
-matplotlib # unpinned
-opencv-python # unpinned , mod_name=cv2
-jax # unpinned, we test for latest version now
-jaxlib # unpinned, we test for latest version now
-tensorflow  # unpinned, we test for latest version now
-torch # unpinned, we test for latest version now
-scipy # unpinned
-dm-haiku # unpinned mod_name=haiku
-flax
-pydriller
+autoflake # for backend generation
 coverage
-scikit-learn # mod_name=sklearn
+dm-haiku # mod_name=haiku
+flax
+h5py
+hypothesis
+jax
+jaxlib
+matplotlib
+networkx
+opencv-python # mod_name=cv2
 pandas
+pydriller
+pymongo
 pyspark
-autoflake # for backend generation
+pytest
+redis
+scikit-learn # mod_name=sklearn
+scipy
 snakeviz # for profiling
+tensorflow
+torch
 torchvision

requirements/requirements.txt

@@ -1,19 +1,20 @@
-numpy # unpinned we test with latest version now
-einops
-psutil
-termcolor
+astor
+astunparse
 colorama
-packaging
-nvidia-ml-py # mod_name=pynvml
+cloudpickle
+cryptography
+dill
 diskcache
+einops
+gast
 google-auth # mod_name=google.auth
-urllib3<2.0 # for version conflict with google-auth
-requests
-pyvis
-dill
-astunparse
 ml-dtypes # mod_name=ml_dtypes
-cloudpickle
-gast
+numpy # unpinned we test with latest version now
+nvidia-ml-py # mod_name=pynvml
+packaging
+psutil
+pyvis
+requests
+termcolor
 tqdm
-cryptography
+urllib3<2.0 # for version conflict with google-auth