use pytorch_lightning in pyproject.toml

openclimatefix · Jul 1, 2024 · 2d55473 · 2d55473
1 parent 50113d2
commit 2d55473
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ dependencies = [
     "ipykernel",
     "h5netcdf",
     "torch>=2.0.0",
-    "lightning==2.3.0",
+    "pytorch_lightning==2.3.0",
     "torchvision",
     "pytest",
     "pytest-cov",