extra

Author: Borda

src/lightning/__setup__.py

@@ -41,7 +41,7 @@ def _prepare_extras() -> dict[str, Any]:
     }
 
     # project specific extras groups
-    extras["fabric-all"] = extras["fabric-strategies"] + extras["fabric-examples"]
+    extras["fabric-all"] = extras["fabric-extra"] + extras["fabric-strategies"] + extras["fabric-examples"]
     extras["fabric-dev"] = extras["fabric-all"] + extras["fabric-test"]
     extras["pytorch-all"] = extras["pytorch-extra"] + extras["pytorch-strategies"] + extras["pytorch-examples"]
     extras["pytorch-dev"] = extras["pytorch-all"] + extras["pytorch-test"]

src/lightning_fabric/__setup__.py

@@ -44,7 +44,7 @@ def _prepare_extras() -> dict[str, Any]:
     }
     for req in parse_requirements(extras["strategies"]):
         extras[req.key] = [str(req)]
-    extras["all"] = extras["strategies"] + extras["examples"]
+    extras["all"] = extras["extra"] + extras["strategies"] + extras["examples"]
     extras["dev"] = extras["all"] + extras["test"]
     return extras