fix

Author: peterdudfield

README.md

@@ -45,7 +45,8 @@ conda activate nowcasting_dataset
 pip install -e .
 ```
 
-Note you can install the [pytorch](https://github.com/pytorch/pytorch) using
+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]
 ```

setup.py

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