Fix torchdiffeq dependency

davidnabergoj · Nov 10, 2024 · 8421dac · 8421dac
1 parent 3179715
commit 8421dac
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/environment.yml b/environment.yml
@@ -6,15 +6,13 @@ dependencies:
   - numpy
   - tqdm
   - pip
+  - pytest  # testing
   - pip:
-    - torchdiffeq
-  # Dependencies for testing and docs
-  - pytest
-  - pip:
-    - Pygments
-    - Babel
-    - sphinx>=8.0.1
-    - sphinx-rtd-theme
-    - sphinx-copybutton
-    - nbsphinx
-    - sphinx-autoapi
+      - torchdiffeq  # mandatory for continuous NFs
+      - Pygments  # docs
+      - Babel  # docs
+      - sphinx>=8.0.1  # docs
+      - sphinx-rtd-theme  # docs
+      - sphinx-copybutton  # docs
+      - nbsphinx  # docs
+      - sphinx-autoapi  # docs