pytorch-lightning 2.5.6__tar.gz → 2.6.0.dev0__tar.gz

{pytorch_lightning-2.5.6/src/pytorch_lightning.egg-info → pytorch_lightning-2.6.0.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytorch-lightning
-Version: 2.5.6
+Version: 2.6.0.dev0
 Summary: PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
 Home-page: https://github.com/Lightning-AI/lightning
 Download-URL: https://github.com/Lightning-AI/lightning
@@ -45,18 +45,19 @@ Requires-Dist: rich>=12.3.0; extra == "extra"
 Requires-Dist: tensorboardX>=2.2; extra == "extra"
 Requires-Dist: bitsandbytes>=0.45.2; platform_system != "Darwin" and extra == "extra"
 Provides-Extra: strategies
-Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
 Provides-Extra: examples
 Requires-Dist: requests<2.33.0; extra == "examples"
 Requires-Dist: torchvision>=0.16.0; extra == "examples"
 Requires-Dist: ipython[all]>=8.0.0; extra == "examples"
 Requires-Dist: torchmetrics>=0.10.0; extra == "examples"
 Provides-Extra: test
-Requires-Dist: coverage==7.10.6; extra == "test"
-Requires-Dist: pytest==8.4.1; extra == "test"
-Requires-Dist: pytest-cov==6.2.1; extra == "test"
+Requires-Dist: coverage==7.10.7; extra == "test"
+Requires-Dist: pytest==8.4.2; extra == "test"
+Requires-Dist: pytest-cov==6.3.0; extra == "test"
 Requires-Dist: pytest-timeout==2.4.0; extra == "test"
-Requires-Dist: pytest-rerunfailures==16.0; extra == "test"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "test"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "test"
 Requires-Dist: pytest-random-order==1.2.0; extra == "test"
 Requires-Dist: cloudpickle>=1.3; extra == "test"
 Requires-Dist: scikit-learn>0.22.1; extra == "test"
@@ -64,13 +65,15 @@ Requires-Dist: numpy>1.20.0; extra == "test"
 Requires-Dist: onnx>1.12.0; extra == "test"
 Requires-Dist: onnxruntime>=1.12.0; extra == "test"
 Requires-Dist: onnxscript>=0.1.0; extra == "test"
-Requires-Dist: psutil<7.0.1; extra == "test"
+Requires-Dist: psutil<7.1.2; extra == "test"
 Requires-Dist: pandas>2.0; extra == "test"
 Requires-Dist: fastapi; extra == "test"
 Requires-Dist: uvicorn; extra == "test"
 Requires-Dist: tensorboard>=2.11; extra == "test"
+Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "test"
+Requires-Dist: huggingface-hub; extra == "test"
 Provides-Extra: deepspeed
-Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "deepspeed"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "deepspeed"
 Provides-Extra: all
 Requires-Dist: matplotlib>3.1; extra == "all"
 Requires-Dist: omegaconf>=2.2.3; extra == "all"
@@ -79,7 +82,7 @@ Requires-Dist: jsonargparse[jsonnet,signatures]>=4.39.0; extra == "all"
 Requires-Dist: rich>=12.3.0; extra == "all"
 Requires-Dist: tensorboardX>=2.2; extra == "all"
 Requires-Dist: bitsandbytes>=0.45.2; 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: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
 Requires-Dist: requests<2.33.0; extra == "all"
 Requires-Dist: torchvision>=0.16.0; extra == "all"
 Requires-Dist: ipython[all]>=8.0.0; extra == "all"
@@ -92,16 +95,17 @@ Requires-Dist: jsonargparse[jsonnet,signatures]>=4.39.0; extra == "dev"
 Requires-Dist: rich>=12.3.0; extra == "dev"
 Requires-Dist: tensorboardX>=2.2; extra == "dev"
 Requires-Dist: bitsandbytes>=0.45.2; platform_system != "Darwin" and extra == "dev"
-Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
+Requires-Dist: deepspeed<0.17.0,>=0.15.0; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
 Requires-Dist: requests<2.33.0; extra == "dev"
 Requires-Dist: torchvision>=0.16.0; extra == "dev"
 Requires-Dist: ipython[all]>=8.0.0; extra == "dev"
 Requires-Dist: torchmetrics>=0.10.0; extra == "dev"
-Requires-Dist: coverage==7.10.6; extra == "dev"
-Requires-Dist: pytest==8.4.1; extra == "dev"
-Requires-Dist: pytest-cov==6.2.1; extra == "dev"
+Requires-Dist: coverage==7.10.7; extra == "dev"
+Requires-Dist: pytest==8.4.2; extra == "dev"
+Requires-Dist: pytest-cov==6.3.0; extra == "dev"
 Requires-Dist: pytest-timeout==2.4.0; extra == "dev"
-Requires-Dist: pytest-rerunfailures==16.0; extra == "dev"
+Requires-Dist: pytest-rerunfailures==16.0.1; python_version < "3.10" and extra == "dev"
+Requires-Dist: pytest-rerunfailures==16.1; python_version >= "3.10" and extra == "dev"
 Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
 Requires-Dist: cloudpickle>=1.3; extra == "dev"
 Requires-Dist: scikit-learn>0.22.1; extra == "dev"
@@ -109,11 +113,13 @@ Requires-Dist: numpy>1.20.0; extra == "dev"
 Requires-Dist: onnx>1.12.0; extra == "dev"
 Requires-Dist: onnxruntime>=1.12.0; extra == "dev"
 Requires-Dist: onnxscript>=0.1.0; extra == "dev"
-Requires-Dist: psutil<7.0.1; extra == "dev"
+Requires-Dist: psutil<7.1.2; extra == "dev"
 Requires-Dist: pandas>2.0; extra == "dev"
 Requires-Dist: fastapi; extra == "dev"
 Requires-Dist: uvicorn; extra == "dev"
 Requires-Dist: tensorboard>=2.11; extra == "dev"
+Requires-Dist: torch-tensorrt; (platform_system == "Linux" and python_version >= "3.12") and extra == "dev"
+Requires-Dist: huggingface-hub; extra == "dev"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -141,7 +147,6 @@ ______________________________________________________________________
 
 <p align="center">
   <a href="https://www.pytorchlightning.ai/">Website</a> •
-  <a href="#advanced-features">Advanced Features</a> •
   <a href="#how-to-use">How To Use</a> •
   <a href="https://lightning.ai/docs/pytorch/stable/">Docs</a> •
   <a href="#examples">Examples</a> •
@@ -159,7 +164,7 @@ ______________________________________________________________________
 [![DockerHub](https://img.shields.io/docker/pulls/pytorchlightning/pytorch_lightning.svg)](https://hub.docker.com/r/pytorchlightning/pytorch_lightning)
 [![codecov](https://codecov.io/gh/Lightning-AI/pytorch-lightning/graph/badge.svg?token=SmzX8mnKlA)](https://codecov.io/gh/Lightning-AI/pytorch-lightning)
 
-[![ReadTheDocs](https://readthedocs.org/projects/pytorch-lightning/badge/?version=2.5.6)](https://lightning.ai/docs/pytorch/stable/)[![Discord](https://img.shields.io/discord/1077906959069626439?style=plastic)](https://discord.gg/VptPCZkGNa)
+[![ReadTheDocs](https://readthedocs.org/projects/pytorch-lightning/badge/?version=2.6.0dev0)](https://lightning.ai/docs/pytorch/stable/)[![Discord](https://img.shields.io/discord/1077906959069626439?style=plastic)](https://discord.gg/VptPCZkGNa)
 [![license](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/Lightning-AI/pytorch-lightning/blob/master/LICENSE)
 
 <!--

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/README.md

@@ -5,7 +5,7 @@
 <br/>
 <br/>
 
-**The deep learning framework to pretrai and finetune AI models.**
+**The deep learning framework to pretrain and finetune AI models.**
 
 **Deploying models?** Check out [LitServe](https://github.com/Lightning-AI/litserve?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme), the PyTorch Lightning for inference engines
 
@@ -64,7 +64,8 @@ Over 340,000 developers use [Lightning Cloud](https://lightning.ai/?utm_source=p
 - [Notebooks](https://lightning.ai/notebooks?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme): Persistent GPU workspaces where AI helps you code and analyze.
 - [Inference](https://lightning.ai/deploy?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme): Deploy models as inference APIs.   
 
-# Why PyTorch Lightning
+<a id="why-pytorch-lightning"></a>
+# Why PyTorch Lightning?   
 
 Training models in plain PyTorch is tedious and error-prone - you have to manually handle things like backprop, mixed precision, multi-GPU, and distributed training, often rewriting code for every new project. PyTorch Lightning organizes PyTorch code to automate those complexities so you can focus on your model and data, while keeping full control and scaling from CPU to multi-node without changing your core code. But if you want control of those things, you can still opt into [expert-level control](#lightning-fabric-expert-control).   
 
@@ -215,7 +216,7 @@ Explore various types of training possible with PyTorch Lightning. Pretrain and
 
 | Task | Description | Run |
 |------|--------------|-----|
-| [Hello world](#pytorch-lightning-example) | Pretrain - Hello world example | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
+| [Hello world](https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Pretrain - Hello world example | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
 | [Image classification](https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - ResNet-34 model to classify images of cars | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
 | [Image segmentation](https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - ResNet-50 model to segment images | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
 | [Object detection](https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme) | Finetune - Faster R-CNN model to detect objects | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning?utm_source=ptl_readme&utm_medium=referral&utm_campaign=ptl_readme"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/pyproject.toml

@@ -91,7 +91,6 @@ ignore = [
 ]
 "src/**" = [
     "S101",  # todo: Use of `assert` detected
-    "S105", "S106", "S107",  # todo: Possible hardcoded password: ...
     "S113",  # todo: Probable use of requests call without timeout
     "S301",  # todo: `pickle` and modules that wrap it can be unsafe when used to deserialize untrusted data, possible security issue
     "S324",  # todo: Probable use of insecure hash functions in `hashlib`
@@ -106,7 +105,6 @@ ignore = [
 ]
 "tests/**" = [
     "S101",  # Use of `assert` detected
-    "S105", "S106",  # todo: Possible hardcoded password: ...
     "S301",  # `pickle` and modules that wrap it can be unsafe when used to deserialize untrusted data, possible security issue
     "S113",  # todo: Probable use of requests call without timeout
     "S311",  # todo: Standard pseudo-random generators are not suitable for cryptographic purposes
@@ -120,7 +118,6 @@ ignore = [
     "S607",  # todo: Starting a process with a partial executable path
     "RET504",  # todo:Unnecessary variable assignment before `return` statement
     "PT004",  # todo: Fixture `tmpdir_unittest_fixture` does not return anything, add leading underscore
-    "PT011",  # todo: `pytest.raises(ValueError)` is too broad, set the `match` parameter or use a more specific exception
     "PT012",  # todo: `pytest.raises()` block should contain a single simple statement
     "PT019",  # todo: Fixture `_` without value is injected as parameter, use `@pytest.mark.usefixtures` instead
 ]

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/requirements/fabric/base.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
 
 torch >=2.1.0, <2.9.0
-fsspec[http] >=2022.5.0, <2025.8.0
+fsspec[http] >=2022.5.0, <2025.11.0
 packaging >=20.0, <=25.0
 typing-extensions >4.5.0, <4.16.0
 lightning-utilities >=0.10.0, <0.16.0

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/requirements/fabric/examples.txt

@@ -1,5 +1,5 @@
 # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
-torchvision >=0.16.0, <0.24.0
+torchvision >=0.16.0, <0.25.0
 torchmetrics >=0.10.0, <1.9.0

pytorch_lightning-2.6.0.dev0/requirements/fabric/extra.txt

@@ -0,0 +1,4 @@
+# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
+#  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
+
+hydra-core >=1.2.0, <1.4.0

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/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.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin"  # strict
+deepspeed >=0.15.0,<0.17.0; platform_system != "Windows" and platform_system != "Darwin"  # strict
 bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"

pytorch_lightning-2.6.0.dev0/requirements/fabric/test.txt

@@ -0,0 +1,12 @@
+coverage ==7.10.7
+numpy >=1.21.0, <1.27.0
+pytest ==8.4.2
+pytest-cov ==6.3.0
+pytest-timeout ==2.4.0
+pytest-rerunfailures ==16.0.1; python_version < "3.10"
+pytest-rerunfailures ==16.1; python_version >= "3.10"
+pytest-random-order ==1.2.0
+click ==8.1.8; python_version < "3.11"
+click ==8.3.0; python_version > "3.10"
+tensorboardX >=2.6, <2.7.0  # todo: relax it back to `>=2.2` after fixing tests
+huggingface-hub

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/requirements/pytorch/base.txt

@@ -4,7 +4,7 @@
 torch >=2.1.0, <2.9.0
 tqdm >=4.57.0, <4.68.0
 PyYAML >5.4, <6.1.0
-fsspec[http] >=2022.5.0, <2025.8.0
+fsspec[http] >=2022.5.0, <2025.11.0
 torchmetrics >0.7.0, <1.9.0
 packaging >=20.0, <=25.0
 typing-extensions >4.5.0, <4.16.0

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/requirements/pytorch/docs.txt

@@ -1,7 +1,7 @@
 -r ../docs.txt
 
 nbformat  # used for generate empty notebook
-ipython[notebook] <9.6.0
+ipython[notebook] <9.7.0
 setuptools<81.0  # workaround for `error in ipython setup command: use_2to3 is invalid.`
 
 onnxscript >= 0.2.2, < 0.5.0

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/requirements/pytorch/examples.txt

@@ -2,6 +2,6 @@
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
 requests <2.33.0
-torchvision >=0.16.0, <0.24.0
-ipython[all] >=8.0.0, <9.0.0
+torchvision >=0.16.0, <0.25.0
+ipython[all] >=8.0.0, <10.0.0
 torchmetrics >=0.10.0, <1.9.0

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/requirements/pytorch/extra.txt

@@ -5,7 +5,7 @@
 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
+jsonargparse[signatures,jsonnet] >=4.39.0, <4.43.0
 rich >=12.3.0, <14.2.0
 tensorboardX >=2.2, <2.7.0  # min version is set by torch.onnx missing attribute
 bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/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.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin"  # strict
+deepspeed >=0.15.0,<0.17.0; platform_system != "Windows" and platform_system != "Darwin"  # strict

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/requirements/pytorch/test.txt

@@ -1,8 +1,9 @@
-coverage ==7.10.6
-pytest ==8.4.1
-pytest-cov ==6.2.1
+coverage ==7.10.7
+pytest ==8.4.2
+pytest-cov ==6.3.0
 pytest-timeout ==2.4.0
-pytest-rerunfailures ==16.0
+pytest-rerunfailures ==16.0.1; python_version < "3.10"
+pytest-rerunfailures ==16.1; python_version >= "3.10"
 pytest-random-order ==1.2.0
 
 # needed in tests
@@ -10,11 +11,14 @@ cloudpickle >=1.3, <3.2.0
 scikit-learn >0.22.1, <1.8.0
 numpy >1.20.0, <1.27.0
 onnx >1.12.0, <1.20.0
-onnxruntime >=1.12.0, <1.23.0
+onnxruntime >=1.12.0, <1.24.0
 onnxscript >= 0.1.0, < 0.5.0
-psutil <7.0.1 # for `DeviceStatsMonitor`
+psutil <7.1.2 # 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.11, <2.21.0  # for `TensorBoardLogger`
+
+torch-tensorrt; platform_system == "Linux" and python_version >= "3.12"
+huggingface-hub

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/src/lightning_fabric/CHANGELOG.md

@@ -4,6 +4,32 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
+
+## [unreleased] - YYYY-MM-DD
+
+### Added
+
+-
+
+
+### Removed
+
+-
+
+
+### Changed
+
+-
+
+
+- Set `_DeviceDtypeModuleMixin._device` from torch's default device function ([#21164](https://github.com/Lightning-AI/pytorch-lightning/pull/21164))
+
+
+### Fixed
+
+- Fixed `EADDRINUSE` errors in distributed tests with port manager and retry logic ([#21309](https://github.com/Lightning-AI/pytorch-lightning/pull/21309))
+
+
 ---
 
 ## [2.5.5] - 2025-09-05
@@ -27,6 +53,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 - Added support for NVIDIA H200 GPUs in `get_available_flops` ([#21119](https://github.com/Lightning-AI/pytorch-lightning/pull/21119))
 
 
+
 ## [2.5.3] - 2025-08-13
 
 ### Changed

{pytorch_lightning-2.5.6 → pytorch_lightning-2.6.0.dev0}/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