Merge pull request #215 from openclimatefix/issue/86-remove-torch

Issue/86 remove torch

Author: peterdudfield

.github/workflows/python-app.yml

@@ -21,7 +21,7 @@ jobs:
           pip install flake8 pytest
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
           pip install git+https://github.com/SheffieldSolar/PV_Live-API
-          pip install -e .
+          pip install -e .[torch]
       - name: Lint with flake8
         run: |
           # stop the build if there are Python syntax errors or undefined names

README.md

@@ -45,6 +45,13 @@ conda activate nowcasting_dataset
 pip install -e .
 ```
 
+Note you can install the [pytorch](https://github.com/pytorch/pytorch)
+and [pytorch_lightning](https://github.com/PyTorchLightning/pytorch-lightning) using
+```shell
+pip install -e .[torch]
+```
+but it is only used to create a dataloader for machine learning models.
+
 ## Pip
 
 A (probably older) version is also available through `pip install nowcasting-dataset`

requirements.txt

@@ -8,8 +8,6 @@ xarray
 ipykernel
 h5netcdf
 gcsfs
-torch
-pytorch-lightning
 dask
 pvlib
 pyproj

setup.py

@@ -17,6 +17,7 @@
     author_email="info@openclimatefix.org",
     company="Open Climate Fix Ltd",
     install_requires=install_requires,
+    extras_require={"torch": ["torch", "pytorch_lightning"]},
     long_description=long_description,
     long_description_content_type="text/markdown",
     packages=find_packages(),