lightning 2.5.3__tar.gz → 2.5.4__tar.gz

{lightning-2.5.3 → lightning-2.5.4}/.actions/assistant.py

@@ -341,47 +341,6 @@ def create_mirror_package(source_dir: str, package_mapping: dict[str, str]) -> N
 
 
 class AssistantCLI:
-    @staticmethod
-    def requirements_prune_pkgs(packages: Sequence[str], req_files: Sequence[str] = REQUIREMENT_FILES_ALL) -> None:
-        """Remove some packages from given requirement files."""
-        if isinstance(req_files, str):
-            req_files = [req_files]
-        for req in req_files:
-            AssistantCLI._prune_packages(req, packages)
-
-    @staticmethod
-    def _prune_packages(req_file: str, packages: Sequence[str]) -> None:
-        """Remove some packages from given requirement files."""
-        path = Path(req_file)
-        assert path.exists()
-        text = path.read_text()
-        lines = text.splitlines()
-        final = []
-        for line in lines:
-            ln_ = line.strip()
-            if not ln_ or ln_.startswith("#"):
-                final.append(line)
-                continue
-            req = list(_parse_requirements([ln_]))[0]
-            if req.name not in packages:
-                final.append(line)
-        print(final)
-        path.write_text("\n".join(final) + "\n")
-
-    @staticmethod
-    def _replace_min(fname: str) -> None:
-        with open(fname, encoding="utf-8") as fopen:
-            req = fopen.read().replace(">=", "==")
-        with open(fname, "w", encoding="utf-8") as fwrite:
-            fwrite.write(req)
-
-    @staticmethod
-    def replace_oldest_ver(requirement_fnames: Sequence[str] = REQUIREMENT_FILES_ALL) -> None:
-        """Replace the min package version by fixed one."""
-        for fname in requirement_fnames:
-            print(fname)
-            AssistantCLI._replace_min(fname)
-
     @staticmethod
     def copy_replace_imports(
         source_dir: str,

{lightning-2.5.3/src/lightning.egg-info → lightning-2.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lightning
-Version: 2.5.3
+Version: 2.5.4
 Summary: The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
 Home-page: https://github.com/Lightning-AI/lightning
 Download-URL: https://github.com/Lightning-AI/lightning
@@ -45,15 +45,15 @@ Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
 Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
 Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
 Provides-Extra: pytorch-strategies
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
 Provides-Extra: pytorch-test
 Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-test"
-Requires-Dist: coverage==7.10.3; extra == "pytorch-test"
+Requires-Dist: coverage==7.10.5; extra == "pytorch-test"
 Requires-Dist: fastapi; extra == "pytorch-test"
-Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-test"
-Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-test"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-test"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-test"
 Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-test"
-Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "pytorch-test"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-test"
 Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-test"
 Requires-Dist: psutil<8.0; extra == "pytorch-test"
 Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-test"
@@ -62,7 +62,7 @@ Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-test"
 Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-test"
 Requires-Dist: pytest==8.4.1; extra == "pytorch-test"
 Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-test"
-Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "pytorch-test"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-test"
 Requires-Dist: uvicorn; extra == "pytorch-test"
 Provides-Extra: pytorch-examples
 Requires-Dist: ipython[all]<9.0; extra == "pytorch-examples"
@@ -71,42 +71,44 @@ Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-examples"
 Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-examples"
 Provides-Extra: fabric-strategies
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-strategies"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
 Provides-Extra: fabric-test
-Requires-Dist: click==8.1.8; extra == "fabric-test"
-Requires-Dist: coverage==7.10.3; extra == "fabric-test"
-Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-test"
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-test"
+Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-test"
+Requires-Dist: coverage==7.10.5; extra == "fabric-test"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-test"
 Requires-Dist: pytest-cov==6.2.1; extra == "fabric-test"
 Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-test"
 Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-test"
 Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-test"
 Requires-Dist: pytest==8.4.1; extra == "fabric-test"
-Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-test"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-test"
 Provides-Extra: fabric-examples
 Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-examples"
 Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-examples"
 Provides-Extra: fabric-all
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-all"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
 Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-all"
 Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-all"
 Provides-Extra: fabric-dev
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-dev"
-Requires-Dist: click==8.1.8; extra == "fabric-dev"
-Requires-Dist: coverage==7.10.3; extra == "fabric-dev"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
-Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-dev"
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-dev"
+Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-dev"
+Requires-Dist: coverage==7.10.5; extra == "fabric-dev"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-dev"
 Requires-Dist: pytest-cov==6.2.1; extra == "fabric-dev"
 Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-dev"
 Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-dev"
 Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-dev"
 Requires-Dist: pytest==8.4.1; extra == "fabric-dev"
-Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-dev"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-dev"
 Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-dev"
 Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-dev"
 Provides-Extra: pytorch-all
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-all"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
 Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-all"
 Requires-Dist: ipython[all]<9.0; extra == "pytorch-all"
 Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-all"
@@ -120,18 +122,18 @@ Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-all"
 Provides-Extra: pytorch-dev
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-dev"
 Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-dev"
-Requires-Dist: coverage==7.10.3; extra == "pytorch-dev"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
+Requires-Dist: coverage==7.10.5; extra == "pytorch-dev"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
 Requires-Dist: fastapi; extra == "pytorch-dev"
 Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-dev"
 Requires-Dist: ipython[all]<9.0; extra == "pytorch-dev"
 Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-dev"
 Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
-Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-dev"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-dev"
 Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-dev"
-Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-dev"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-dev"
 Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-dev"
-Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "pytorch-dev"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-dev"
 Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-dev"
 Requires-Dist: psutil<8.0; extra == "pytorch-dev"
 Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-dev"
@@ -142,7 +144,7 @@ Requires-Dist: pytest==8.4.1; extra == "pytorch-dev"
 Requires-Dist: requests<3.0; extra == "pytorch-dev"
 Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-dev"
 Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-dev"
-Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "pytorch-dev"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-dev"
 Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-dev"
 Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-dev"
 Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-dev"
@@ -157,16 +159,18 @@ Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
 Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
 Provides-Extra: strategies
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "strategies"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
 Provides-Extra: test
-Requires-Dist: click==8.1.8; extra == "test"
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "test"
+Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "test"
 Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
-Requires-Dist: coverage==7.10.3; extra == "test"
+Requires-Dist: coverage==7.10.5; extra == "test"
 Requires-Dist: fastapi; extra == "test"
-Requires-Dist: numpy<2.0,>=1.17.2; extra == "test"
-Requires-Dist: onnx<2.0,>=1.12.0; extra == "test"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "test"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "test"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "test"
 Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "test"
-Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "test"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "test"
 Requires-Dist: pandas<3.0,>2.0; extra == "test"
 Requires-Dist: psutil<8.0; extra == "test"
 Requires-Dist: pytest-cov==6.2.1; extra == "test"
@@ -175,8 +179,8 @@ Requires-Dist: pytest-rerunfailures==15.1; extra == "test"
 Requires-Dist: pytest-timeout==2.4.0; extra == "test"
 Requires-Dist: pytest==8.4.1; extra == "test"
 Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "test"
-Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "test"
-Requires-Dist: tensorboardX<3.0,>=2.2; extra == "test"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "test"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "test"
 Requires-Dist: uvicorn; extra == "test"
 Provides-Extra: examples
 Requires-Dist: ipython[all]<9.0; extra == "examples"
@@ -187,7 +191,7 @@ Provides-Extra: data
 Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
 Provides-Extra: all
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "all"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
 Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "all"
 Requires-Dist: ipython[all]<9.0; extra == "all"
 Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "all"
@@ -200,20 +204,22 @@ Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "all"
 Requires-Dist: torchvision<1.0,>=0.16.0; extra == "all"
 Provides-Extra: dev
 Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "dev"
-Requires-Dist: click==8.1.8; extra == "dev"
+Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "dev"
+Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "dev"
 Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
-Requires-Dist: coverage==7.10.3; extra == "dev"
-Requires-Dist: deepspeed<=0.9.3,>=0.9.3; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
+Requires-Dist: coverage==7.10.5; extra == "dev"
+Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
 Requires-Dist: fastapi; extra == "dev"
 Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "dev"
 Requires-Dist: ipython[all]<9.0; extra == "dev"
 Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "dev"
 Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
-Requires-Dist: numpy<2.0,>=1.17.2; extra == "dev"
+Requires-Dist: numpy<2.0,>1.20.0; extra == "dev"
+Requires-Dist: numpy<2.0,>=1.21.0; extra == "dev"
 Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "dev"
-Requires-Dist: onnx<2.0,>=1.12.0; extra == "dev"
+Requires-Dist: onnx<2.0,>1.12.0; extra == "dev"
 Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "dev"
-Requires-Dist: onnxscript<1.0,>=0.2.2; extra == "dev"
+Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "dev"
 Requires-Dist: pandas<3.0,>2.0; extra == "dev"
 Requires-Dist: psutil<8.0; extra == "dev"
 Requires-Dist: pytest-cov==6.2.1; extra == "dev"
@@ -224,8 +230,9 @@ Requires-Dist: pytest==8.4.1; extra == "dev"
 Requires-Dist: requests<3.0; extra == "dev"
 Requires-Dist: rich<15.0,>=12.3.0; extra == "dev"
 Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "dev"
-Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "dev"
+Requires-Dist: tensorboard<3.0,>=2.11; extra == "dev"
 Requires-Dist: tensorboardX<3.0,>=2.2; extra == "dev"
+Requires-Dist: tensorboardX<3.0,>=2.6; extra == "dev"
 Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "dev"
 Requires-Dist: torchvision<1.0,>=0.16.0; extra == "dev"
 Requires-Dist: uvicorn; extra == "dev"
@@ -265,7 +272,7 @@ ______________________________________________________________________
   <a href="#lightning-fabric-expert-control">Fabric</a> •
   <a href="https://lightning.ai/">Lightning AI</a> •   
   <a href="#community">Community</a> •
-  <a href="https://pytorch-lightning.readthedocs.io/en/2.5.3">Docs</a>
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.5.4">Docs</a>
 </p>
 
 <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -796,7 +803,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
 
 |       System / PyTorch ver.        | 1.13                                                                                                                                                                                                                            | 2.0                                                                                                                                                                                                                             |                                                                                                               2.1                                                                                                               |
 | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
-|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.5.3)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
+|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.5.4)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
 |  Linux (multiple Python versions)  | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |
 |   OSX (multiple Python versions)   | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |
 | Windows (multiple Python versions) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |

{lightning-2.5.3 → lightning-2.5.4}/pyproject.toml

@@ -179,6 +179,7 @@ markers = [
     "cloud: Run the cloud tests for example",
 ]
 filterwarnings = [
+    # "error::DeprecationWarning",
     "error::FutureWarning",
     "ignore::FutureWarning:onnxscript", # Temporary ignore until onnxscript is updated
 ]

{lightning-2.5.3 → lightning-2.5.4}/requirements/fabric/strategies.txt

@@ -5,5 +5,5 @@
 
 # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
 #  shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
-deepspeed >=0.9.3, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin"  # strict
+deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin"  # strict
 bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"

lightning-2.5.4/requirements/fabric/test.txt

@@ -0,0 +1,10 @@
+coverage ==7.10.5
+numpy >=1.21.0, <1.27.0
+pytest ==8.4.1
+pytest-cov ==6.2.1
+pytest-timeout ==2.4.0
+pytest-rerunfailures ==15.1
+pytest-random-order ==1.2.0
+click ==8.1.8; python_version < "3.11"
+click ==8.2.1; python_version > "3.10"
+tensorboardX >=2.6, <2.7.0  # todo: relax it back to `>=2.2` after fixing tests

{lightning-2.5.3 → lightning-2.5.4}/requirements/pytorch/docs.txt

@@ -1,9 +1,9 @@
 -r ../docs.txt
 
 nbformat  # used for generate empty notebook
-ipython[notebook] <8.19.0
+ipython[notebook] <9.5.0
 setuptools<81.0  # workaround for `error in ipython setup command: use_2to3 is invalid.`
 
-onnxscript >= 0.2.2, <0.4.0
+onnxscript >= 0.2.2, < 0.5.0
 
 #-r ../../_notebooks/.actions/requires.txt

{lightning-2.5.3 → lightning-2.5.4}/requirements/pytorch/extra.txt

@@ -2,7 +2,7 @@
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
 # extended list of package dependencies to reach full functionality
-matplotlib>3.1, <3.10.0
+matplotlib>3.1, <3.11.0
 omegaconf >=2.2.3, <2.4.0
 hydra-core >=1.2.0, <1.4.0
 jsonargparse[signatures,jsonnet] >=4.39.0, <4.41.0

{lightning-2.5.3 → lightning-2.5.4}/requirements/pytorch/strategies.txt

@@ -3,4 +3,4 @@
 
 # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
 #  shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
-deepspeed >=0.9.3, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin"  # strict
+deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin"  # strict

{lightning-2.5.3 → lightning-2.5.4}/requirements/pytorch/test.txt

@@ -1,4 +1,4 @@
-coverage ==7.10.3
+coverage ==7.10.5
 pytest ==8.4.1
 pytest-cov ==6.2.1
 pytest-timeout ==2.4.0
@@ -7,14 +7,14 @@ pytest-random-order ==1.2.0
 
 # needed in tests
 cloudpickle >=1.3, <3.2.0
-scikit-learn >0.22.1, <1.7.0
-numpy >=1.17.2, <1.27.0
-onnx >=1.12.0, <1.19.0
-onnxruntime >=1.12.0, <1.21.0
-onnxscript >= 0.2.2, <0.4.0
+scikit-learn >0.22.1, <1.8.0
+numpy >1.20.0, <1.27.0
+onnx >1.12.0, <1.19.0
+onnxruntime >=1.12.0, <1.23.0
+onnxscript >= 0.1.0, < 0.5.0
 psutil <7.0.1 # for `DeviceStatsMonitor`
 pandas >2.0, <2.4.0  # needed in benchmarks
 fastapi  # for `ServableModuleValidator`  # not setting version as re-defined in App
 uvicorn  # for `ServableModuleValidator`  # not setting version as re-defined in App
 
-tensorboard >=2.9.1, <2.21.0  # for `TensorBoardLogger`
+tensorboard >=2.11, <2.21.0  # for `TensorBoardLogger`

{lightning-2.5.3 → lightning-2.5.4}/src/lightning/fabric/CHANGELOG.md

@@ -6,7 +6,14 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ---
 
-## [2.5.3] - 2025-08-DD
+## [2.5.4] - 2025-08-29
+
+### Changed
+
+- Added support for NVIDIA H200 GPUs in `get_available_flops` ([#20913](https://github.com/Lightning-AI/pytorch-lightning/pull/21119))
+
+
+## [2.5.3] - 2025-08-13
 
 ### Changed
 

{lightning-2.5.3 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/torch_collective.py

@@ -24,6 +24,8 @@ class TorchCollective(Collective):
     """
 
     manages_default_group = False
+    addr_key = "MASTER_ADDR"
+    port_key = "MASTER_PORT"
 
     def __init__(self) -> None:
         if not dist.is_available():
@@ -136,26 +138,21 @@ class TorchCollective(Collective):
         if self.is_initialized():
             return self
         # maybe set addr
-        set_addr = False
-        addr_key = "MASTER_ADDR"
-        if main_address is not None and addr_key not in os.environ:
-            os.environ[addr_key] = main_address
-            set_addr = True
+        setting_env = []
+        if main_address is not None and self.addr_key not in os.environ:
+            os.environ[self.addr_key] = main_address
+            setting_env.append(self.addr_key)
         # maybe set port
-        set_port = False
-        port_key = "MASTER_PORT"
-        if main_port is not None and port_key not in os.environ:
-            os.environ[port_key] = str(main_port)
-            set_port = True
+        if main_port is not None and self.port_key not in os.environ:
+            os.environ[self.port_key] = str(main_port)
+            setting_env.append(self.port_key)
         # this will `init_group`
         super().setup(**kwargs)
         # set as a class attribute so any instance can know whether we initialized the default process group
         TorchCollective.manages_default_group = True
         # cleanup
-        if set_addr:
-            os.environ.pop("MASTER_ADDR", None)
-        if set_port:
-            os.environ.pop("MASTER_PORT", None)
+        for kenv in setting_env:
+            os.environ.pop(kenv, None)
         return self
 
     @override

{lightning-2.5.3 → lightning-2.5.4}/src/lightning/fabric/strategies/deepspeed.py

@@ -46,7 +46,6 @@ if TYPE_CHECKING:
     from deepspeed import DeepSpeedEngine
 
 _DEEPSPEED_AVAILABLE = RequirementCache("deepspeed")
-_DEEPSPEED_GREATER_EQUAL_0_14_1 = RequirementCache("deepspeed>=0.14.1")
 
 
 # TODO(fabric): Links in the docstrings to PL-specific deepspeed user docs need to be replaced.
@@ -503,10 +502,7 @@ class DeepSpeedStrategy(DDPStrategy, _Sharded):
             )
         engine = engines[0]
 
-        if _DEEPSPEED_GREATER_EQUAL_0_14_1:
-            from deepspeed.runtime.base_optimizer import DeepSpeedOptimizer
-        else:
-            from deepspeed.runtime import DeepSpeedOptimizer
+        from deepspeed.runtime.base_optimizer import DeepSpeedOptimizer
 
         optimzer_state_requested = any(isinstance(item, (Optimizer, DeepSpeedOptimizer)) for item in state.values())
 

{lightning-2.5.3 → lightning-2.5.4}/src/lightning/fabric/utilities/throughput.py

@@ -304,6 +304,23 @@ def measure_flops(
 
 _CUDA_FLOPS: dict[str, dict[Union[str, torch.dtype], float]] = {
     # Hopper
+    # source: https://nvdam.widen.net/s/nb5zzzsjdf/hpc-datasheet-sc23-h200-datasheet-3002446
+    "h200 sxm1": {
+        torch.float64: 3.4e13,
+        torch.float32: 6.7e13,
+        "tfloat32": 9.9e14,
+        torch.bfloat16: 2.0e15,
+        torch.float16: 2.0e15,
+        torch.int8: 4.0e15,
+    },
+    "h200 nvl1": {
+        torch.float64: 3.0e13,
+        torch.float32: 6.0e13,
+        "tfloat32": 8.4e14,
+        torch.bfloat16: 1.7e15,
+        torch.float16: 1.7e15,
+        torch.int8: 3.3e15,
+    },
     # source: https://resources.nvidia.com/en-us-tensor-core
     "h100 nvl": {
         torch.float64: 67e12,
@@ -536,7 +553,12 @@ def get_available_flops(device: torch.device, dtype: Union[torch.dtype, str]) ->
     if device.type == "cuda":
         device_name = torch.cuda.get_device_name(device)
         chip = device_name.lower()
-        if "h100" in chip:
+        if "h200" in chip:
+            if "sxm1" in chip:
+                chip = "h200 sxm1"
+            elif "nvl1" in chip:
+                chip = "h200 nvl1"
+        elif "h100" in chip:
             if "hbm3" in chip:
                 chip = "h100 sxm"
             elif "nvl" in chip:

{lightning-2.5.3 → lightning-2.5.4}/src/lightning/pytorch/CHANGELOG.md

@@ -6,7 +6,18 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ---
 
-## [2.5.3] - 2025-08-DD
+## [2.5.4] - 2025-08-29
+
+### Fixed
+
+- Fixed `AsyncCheckpointIO` snapshots tensors to avoid race with parameter mutation ([#21079](https://github.com/Lightning-AI/pytorch-lightning/pull/21079))
+- Fixed `AsyncCheckpointIO` threadpool exception if calling fit or validate more than one ([#20952](https://github.com/Lightning-AI/pytorch-lightning/pull/20952))
+- Fixed learning rate not being correctly set after using `LearningRateFinder` callback ([#21068](https://github.com/Lightning-AI/pytorch-lightning/pull/21068))
+- Fixed misalignment column while using rich model summary in `DeepSpeedstrategy` ([#21100](https://github.com/Lightning-AI/pytorch-lightning/pull/21100))
+- Fixed `RichProgressBar` crashing when sanity checking using val dataloader with 0 len ([#21108](https://github.com/Lightning-AI/pytorch-lightning/pull/21108))
+
+
+## [2.5.3] - 2025-08-13
 
 ### Changed
 
@@ -57,7 +68,6 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 - Allow LightningCLI to use a customized argument parser class ([#20596](https://github.com/Lightning-AI/pytorch-lightning/pull/20596))
 - Change `wandb` default x-axis to `tensorboard`'s `global_step` when `sync_tensorboard=True` ([#20611](https://github.com/Lightning-AI/pytorch-lightning/pull/20611))
-- Added a new `checkpoint_path_prefix` parameter to the MLflow logger which can control the path to where the MLflow artifacts for the model checkpoints are stored ([#20538](https://github.com/Lightning-AI/pytorch-lightning/pull/20538))
 - CometML logger was updated to support the recent Comet SDK ([#20275](https://github.com/Lightning-AI/pytorch-lightning/pull/20275))
 - bump: testing with latest `torch` 2.6 ([#20509](https://github.com/Lightning-AI/pytorch-lightning/pull/20509))
 

{lightning-2.5.3 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/rich_progress.py

@@ -387,8 +387,7 @@ class RichProgressBar(ProgressBar):
 
     @override
     def on_sanity_check_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule") -> None:
-        if self.progress is not None:
-            assert self.val_sanity_progress_bar_id is not None
+        if self.progress is not None and self.val_sanity_progress_bar_id is not None:
             self.progress.update(self.val_sanity_progress_bar_id, advance=0, visible=False)
         self.refresh()
 

{lightning-2.5.3 → lightning-2.5.4}/src/lightning/pytorch/callbacks/rich_model_summary.py

@@ -78,6 +78,7 @@ class RichModelSummary(ModelSummary):
         from rich.table import Table
 
         console = get_console()
+        column_names = list(zip(*summary_data))[0]
 
         header_style: str = summarize_kwargs.get("header_style", "bold magenta")
         table = Table(header_style=header_style)
@@ -85,9 +86,11 @@ class RichModelSummary(ModelSummary):
         table.add_column("Name", justify="left", no_wrap=True)
         table.add_column("Type")
         table.add_column("Params", justify="right")
-        table.add_column("Mode")
 
-        column_names = list(zip(*summary_data))[0]
+        if "Params per Device" in column_names:
+            table.add_column("Params per Device", justify="right")
+
+        table.add_column("Mode")
 
         for column_name in ["In sizes", "Out sizes"]:
             if column_name in column_names: