Update requirements

environment.yaml

@@ -47,14 +47,9 @@ dependencies:
       - normflows
       - nflows
       - einops
-      - ipdb
       - torchsde
       - torchdiffeq
       - torchcfm
-      - black
-      - flake8
-      - isort
-      - jupyter-black
       - git+https://github.com/VincentStimper/resampled-base-flows.git # larsflow
       - git+https://github.com/noegroup/bgflow.git
       - git+https://github.com/jarridrb/fab-torch.git

requirements.txt

@@ -5,9 +5,9 @@ lightning>=2.0.0
 torchmetrics>=0.11.4
 
 # --------- hydra --------- #
-hydra-core==1.3.2
-hydra-colorlog==1.2.0
-hydra-optuna-sweeper==1.2.0
+hydra-core>=1.0
+hydra-colorlog
+hydra-optuna-sweeper
 
 # --------- loggers --------- #
 # wandb
@@ -22,3 +22,18 @@ pre-commit      # hooks for applying linters on commit
 rich            # beautiful text formatting in terminal
 pytest          # tests
 # sh            # for running bash commands in some tests (linux/macos only)
+
+torchdiffeq
+torchcfm
+
+black
+flake8
+isort
+einops
+nflows
+normflows
+git+https://github.com/VincentStimper/resampled-base-flows.git # larsflow
+git+https://github.com/noegroup/bgflow.git
+git+https://github.com/jarridrb/fab-torch.git
+scikit-learn
+scipy