Update requirements.

simonpf · May 20, 2024 · c5c26a6 · c5c26a6
1 parent 3324145
commit c5c26a6
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/.readthedocs.yml b/.readthedocs.yml
@@ -13,7 +13,7 @@ python:
     - method: pip
       path: .
       extra_requirements:
-        - sphinx
+        - "docs"
 
 sphinx:
   builder: html

diff --git a/gprof_nn/training.py b/gprof_nn/training.py
@@ -13,6 +13,7 @@
 
 
 from gprof_nn import sensors
+import gprof_nn.logging
 
 
 LOGGER = logging.Logger(__name__)

diff --git a/setup.py b/setup.py
@@ -19,13 +19,16 @@
     author_email="simon.pfreundschuh@chalmers.se",
     install_requires=[
         "numpy", "scipy", "xarray", "torch", "appdirs", "rich",
-        "quantnn>=0.0.5dev", "h5py", "netCDF4", "h5netcdf", "pandas"
+        "quantnn>=0.0.5dev", "h5py", "netCDF4", "h5netcdf", "pandas",
     ],
     extras_require = {
         'development': [
             'pytest', 'pykdtree'
-            ]
-        },
+            ],
+        'docs': [
+            "sphinx", "jupyter-book"
+        ]
+    },
     entry_points = {
         'console_scripts': ['gprof_nn=gprof_nn.cli:gprof_nn'],
     },