lightning 2.5.2__tar.gz → 2.5.4__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (300) hide show
  1. {lightning-2.5.2 → lightning-2.5.4}/.actions/assistant.py +0 -41
  2. {lightning-2.5.2/src/lightning.egg-info → lightning-2.5.4}/PKG-INFO +88 -80
  3. {lightning-2.5.2 → lightning-2.5.4}/README.md +6 -0
  4. {lightning-2.5.2 → lightning-2.5.4}/pyproject.toml +2 -0
  5. {lightning-2.5.2 → lightning-2.5.4}/requirements/base.txt +3 -3
  6. {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/base.txt +4 -4
  7. {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/examples.txt +2 -3
  8. {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/strategies.txt +1 -1
  9. lightning-2.5.4/requirements/fabric/test.txt +10 -0
  10. {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/base.txt +6 -6
  11. {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/docs.txt +4 -2
  12. {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/examples.txt +2 -3
  13. {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/extra.txt +2 -2
  14. {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/strategies.txt +1 -1
  15. {lightning-2.5.2 → lightning-2.5.4}/requirements/pytorch/test.txt +9 -8
  16. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__about__.py +1 -1
  17. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/CHANGELOG.md +20 -0
  18. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/registry.py +2 -2
  19. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/cli.py +10 -2
  20. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/torch_collective.py +11 -14
  21. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/xla.py +15 -0
  22. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/bitsandbytes.py +5 -3
  23. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/deepspeed.py +1 -5
  24. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/consolidate_checkpoint.py +6 -5
  25. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/device_parser.py +15 -0
  26. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/imports.py +2 -1
  27. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/seed.py +4 -5
  28. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/spike.py +1 -4
  29. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/testing/_runif.py +7 -5
  30. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/throughput.py +23 -1
  31. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/CHANGELOG.md +35 -1
  32. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/lr_finder.py +1 -1
  33. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/model_checkpoint.py +49 -6
  34. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/rich_progress.py +40 -7
  35. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/rich_model_summary.py +5 -2
  36. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/module.py +62 -11
  37. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/optimizer.py +1 -1
  38. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/transformer.py +13 -4
  39. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/loop.py +8 -0
  40. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/manual.py +2 -1
  41. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/training_epoch_loop.py +5 -1
  42. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/async_plugin.py +40 -5
  43. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/advanced.py +6 -4
  44. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/call.py +2 -1
  45. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +6 -11
  46. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/data_connector.py +8 -0
  47. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +3 -2
  48. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +1 -1
  49. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/trainer.py +3 -4
  50. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/lr_finder.py +26 -33
  51. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/imports.py +4 -1
  52. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_summary/model_summary.py +17 -1
  53. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +1 -0
  54. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/parsing.py +2 -1
  55. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/testing/_runif.py +11 -4
  56. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +3 -2
  57. lightning-2.5.4/src/lightning/version.info +1 -0
  58. {lightning-2.5.2 → lightning-2.5.4/src/lightning.egg-info}/PKG-INFO +88 -80
  59. {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/SOURCES.txt +1 -1
  60. {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/requires.txt +78 -60
  61. lightning-2.5.4/src/version.info +1 -0
  62. lightning-2.5.2/requirements/fabric/test.txt +0 -9
  63. lightning-2.5.2/src/lightning/version.info +0 -1
  64. lightning-2.5.2/src/version.info +0 -1
  65. {lightning-2.5.2 → lightning-2.5.4}/CITATION.cff +0 -0
  66. {lightning-2.5.2 → lightning-2.5.4}/LICENSE +0 -0
  67. {lightning-2.5.2 → lightning-2.5.4}/requirements/fabric/docs.txt +0 -0
  68. {lightning-2.5.2 → lightning-2.5.4}/setup.cfg +0 -0
  69. {lightning-2.5.2 → lightning-2.5.4}/setup.py +0 -0
  70. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__init__.py +0 -0
  71. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__setup__.py +0 -0
  72. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/__version__.py +0 -0
  73. /lightning-2.5.2/src/lightning/ffd6261c9058703d8c54093b4f93e2b4008030713613e504 → /lightning-2.5.4/src/lightning/d4cfc24c31e95ba11b2250aa0b5d0d903d59b77c4fbee1d2 +0 -0
  74. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/data/__init__.py +0 -0
  75. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/__init__.py +0 -0
  76. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/_graveyard/__init__.py +0 -0
  77. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/_graveyard/tpu.py +0 -0
  78. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/__init__.py +0 -0
  79. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/accelerator.py +0 -0
  80. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/cpu.py +0 -0
  81. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/cuda.py +0 -0
  82. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/mps.py +0 -0
  83. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/accelerators/xla.py +0 -0
  84. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/connector.py +0 -0
  85. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/fabric.py +0 -0
  86. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/__init__.py +0 -0
  87. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/csv_logs.py +0 -0
  88. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/logger.py +0 -0
  89. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/loggers/tensorboard.py +0 -0
  90. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/__init__.py +0 -0
  91. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
  92. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
  93. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/collectives/single_device.py +0 -0
  94. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
  95. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
  96. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
  97. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/lightning.py +0 -0
  98. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/lsf.py +0 -0
  99. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
  100. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
  101. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
  102. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/__init__.py +0 -0
  103. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
  104. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/torch_io.py +0 -0
  105. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/io/xla.py +0 -0
  106. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
  107. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/amp.py +0 -0
  108. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/deepspeed.py +0 -0
  109. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/double.py +0 -0
  110. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/fsdp.py +0 -0
  111. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/half.py +0 -0
  112. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/precision.py +0 -0
  113. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/transformer_engine.py +0 -0
  114. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/utils.py +0 -0
  115. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/plugins/precision/xla.py +0 -0
  116. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/__init__.py +0 -0
  117. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/ddp.py +0 -0
  118. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/dp.py +0 -0
  119. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/fsdp.py +0 -0
  120. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
  121. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
  122. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/multiprocessing.py +0 -0
  123. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/subprocess_script.py +0 -0
  124. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
  125. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/model_parallel.py +0 -0
  126. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/parallel.py +0 -0
  127. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/registry.py +0 -0
  128. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/single_device.py +0 -0
  129. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/single_xla.py +0 -0
  130. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/strategy.py +0 -0
  131. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/xla.py +0 -0
  132. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/strategies/xla_fsdp.py +0 -0
  133. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/__init__.py +0 -0
  134. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/apply_func.py +0 -0
  135. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/cloud_io.py +0 -0
  136. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/data.py +0 -0
  137. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/device_dtype_mixin.py +0 -0
  138. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/distributed.py +0 -0
  139. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/enums.py +0 -0
  140. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/exceptions.py +0 -0
  141. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/init.py +0 -0
  142. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/load.py +0 -0
  143. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/logger.py +0 -0
  144. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/optimizer.py +0 -0
  145. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/rank_zero.py +0 -0
  146. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/registry.py +0 -0
  147. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
  148. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/types.py +0 -0
  149. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/utilities/warnings.py +0 -0
  150. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/fabric/wrappers.py +0 -0
  151. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/py.typed +0 -0
  152. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/__init__.py +0 -0
  153. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
  154. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
  155. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
  156. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/precision.py +0 -0
  157. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
  158. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/__init__.py +0 -0
  159. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
  160. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/cpu.py +0 -0
  161. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/cuda.py +0 -0
  162. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/mps.py +0 -0
  163. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/accelerators/xla.py +0 -0
  164. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/__init__.py +0 -0
  165. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
  166. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/callback.py +0 -0
  167. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
  168. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/device_stats_monitor.py +0 -0
  169. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/early_stopping.py +0 -0
  170. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/finetuning.py +0 -0
  171. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +0 -0
  172. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
  173. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/lr_monitor.py +0 -0
  174. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/model_summary.py +0 -0
  175. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
  176. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/prediction_writer.py +0 -0
  177. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
  178. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/progress_bar.py +0 -0
  179. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +0 -0
  180. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/pruning.py +0 -0
  181. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/spike.py +0 -0
  182. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +0 -0
  183. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/throughput_monitor.py +0 -0
  184. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/callbacks/timer.py +0 -0
  185. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/cli.py +0 -0
  186. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/__init__.py +0 -0
  187. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/datamodule.py +0 -0
  188. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/hooks.py +0 -0
  189. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
  190. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/mixins/hparams_mixin.py +0 -0
  191. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/core/saving.py +0 -0
  192. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/__init__.py +0 -0
  193. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/boring_classes.py +0 -0
  194. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/lstm.py +0 -0
  195. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
  196. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/__init__.py +0 -0
  197. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/comet.py +0 -0
  198. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/csv_logs.py +0 -0
  199. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/logger.py +0 -0
  200. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/mlflow.py +0 -0
  201. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/neptune.py +0 -0
  202. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/tensorboard.py +0 -0
  203. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/utilities.py +0 -0
  204. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loggers/wandb.py +0 -0
  205. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/__init__.py +0 -0
  206. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/evaluation_loop.py +0 -0
  207. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/fetchers.py +0 -0
  208. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/fit_loop.py +0 -0
  209. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
  210. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/automatic.py +0 -0
  211. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
  212. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/prediction_loop.py +0 -0
  213. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/progress.py +0 -0
  214. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/loops/utilities.py +0 -0
  215. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/overrides/__init__.py +0 -0
  216. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/overrides/distributed.py +0 -0
  217. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/__init__.py +0 -0
  218. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
  219. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
  220. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
  221. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
  222. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/wrapper.py +0 -0
  223. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
  224. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
  225. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
  226. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/amp.py +0 -0
  227. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +0 -0
  228. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/deepspeed.py +0 -0
  229. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/double.py +0 -0
  230. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/fsdp.py +0 -0
  231. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/half.py +0 -0
  232. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/precision.py +0 -0
  233. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
  234. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
  235. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/__init__.py +0 -0
  236. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/base.py +0 -0
  237. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/profiler.py +0 -0
  238. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/pytorch.py +0 -0
  239. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/simple.py +0 -0
  240. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/profilers/xla.py +0 -0
  241. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/serve/__init__.py +0 -0
  242. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/serve/servable_module.py +0 -0
  243. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/serve/servable_module_validator.py +0 -0
  244. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/__init__.py +0 -0
  245. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/ddp.py +0 -0
  246. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/deepspeed.py +0 -0
  247. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/fsdp.py +0 -0
  248. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
  249. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
  250. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +0 -0
  251. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +0 -0
  252. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
  253. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/model_parallel.py +0 -0
  254. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/parallel.py +0 -0
  255. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/single_device.py +0 -0
  256. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/single_xla.py +0 -0
  257. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/strategy.py +0 -0
  258. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/strategies/xla.py +0 -0
  259. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/__init__.py +0 -0
  260. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
  261. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
  262. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/callback_connector.py +0 -0
  263. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +0 -0
  264. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
  265. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
  266. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/connectors/signal_connector.py +0 -0
  267. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/setup.py +0 -0
  268. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/trainer/states.py +0 -0
  269. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/__init__.py +0 -0
  270. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/batch_size_scaling.py +0 -0
  271. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/tuner/tuning.py +0 -0
  272. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/__init__.py +0 -0
  273. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/_pytree.py +0 -0
  274. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/argparse.py +0 -0
  275. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/combined_loader.py +0 -0
  276. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/compile.py +0 -0
  277. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +0 -0
  278. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/data.py +0 -0
  279. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/deepspeed.py +0 -0
  280. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/enums.py +0 -0
  281. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/exceptions.py +0 -0
  282. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/grads.py +0 -0
  283. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/memory.py +0 -0
  284. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
  285. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
  286. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/migration/utils.py +0 -0
  287. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_helpers.py +0 -0
  288. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_registry.py +0 -0
  289. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
  290. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
  291. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
  292. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/seed.py +0 -0
  293. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
  294. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
  295. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/types.py +0 -0
  296. {lightning-2.5.2 → lightning-2.5.4}/src/lightning/pytorch/utilities/warnings.py +0 -0
  297. {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/dependency_links.txt +0 -0
  298. {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/entry_points.txt +0 -0
  299. {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/not-zip-safe +0 -0
  300. {lightning-2.5.2 → lightning-2.5.4}/src/lightning.egg-info/top_level.txt +0 -0
@@ -341,47 +341,6 @@ def create_mirror_package(source_dir: str, package_mapping: dict[str, str]) -> N
341
341
 
342
342
 
343
343
  class AssistantCLI:
344
- @staticmethod
345
- def requirements_prune_pkgs(packages: Sequence[str], req_files: Sequence[str] = REQUIREMENT_FILES_ALL) -> None:
346
- """Remove some packages from given requirement files."""
347
- if isinstance(req_files, str):
348
- req_files = [req_files]
349
- for req in req_files:
350
- AssistantCLI._prune_packages(req, packages)
351
-
352
- @staticmethod
353
- def _prune_packages(req_file: str, packages: Sequence[str]) -> None:
354
- """Remove some packages from given requirement files."""
355
- path = Path(req_file)
356
- assert path.exists()
357
- text = path.read_text()
358
- lines = text.splitlines()
359
- final = []
360
- for line in lines:
361
- ln_ = line.strip()
362
- if not ln_ or ln_.startswith("#"):
363
- final.append(line)
364
- continue
365
- req = list(_parse_requirements([ln_]))[0]
366
- if req.name not in packages:
367
- final.append(line)
368
- print(final)
369
- path.write_text("\n".join(final) + "\n")
370
-
371
- @staticmethod
372
- def _replace_min(fname: str) -> None:
373
- with open(fname, encoding="utf-8") as fopen:
374
- req = fopen.read().replace(">=", "==")
375
- with open(fname, "w", encoding="utf-8") as fwrite:
376
- fwrite.write(req)
377
-
378
- @staticmethod
379
- def replace_oldest_ver(requirement_fnames: Sequence[str] = REQUIREMENT_FILES_ALL) -> None:
380
- """Replace the min package version by fixed one."""
381
- for fname in requirement_fnames:
382
- print(fname)
383
- AssistantCLI._replace_min(fname)
384
-
385
344
  @staticmethod
386
345
  def copy_replace_imports(
387
346
  source_dir: str,
@@ -1,11 +1,11 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lightning
3
- Version: 2.5.2
3
+ Version: 2.5.4
4
4
  Summary: The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
5
5
  Home-page: https://github.com/Lightning-AI/lightning
6
6
  Download-URL: https://github.com/Lightning-AI/lightning
7
7
  Author: Lightning AI et al.
8
- Author-email: pytorch@lightning.ai
8
+ Author-email: developer@lightning.ai
9
9
  License: Apache-2.0
10
10
  Project-URL: Bug Tracker, https://github.com/Lightning-AI/pytorch-lightning/issues
11
11
  Project-URL: Documentation, https://lightning.ai/lightning-docs
@@ -27,93 +27,91 @@ Classifier: Programming Language :: Python :: 3.12
27
27
  Requires-Python: >=3.9
28
28
  Description-Content-Type: text/markdown
29
29
  License-File: LICENSE
30
- Requires-Dist: PyYAML<8.0,>=5.4
30
+ Requires-Dist: PyYAML<8.0,>5.4
31
31
  Requires-Dist: fsspec[http]<2027.0,>=2022.5.0
32
32
  Requires-Dist: lightning-utilities<2.0,>=0.10.0
33
33
  Requires-Dist: packaging<27.0,>=20.0
34
34
  Requires-Dist: torch<4.0,>=2.1.0
35
- Requires-Dist: torchmetrics<3.0,>=0.7.0
35
+ Requires-Dist: torchmetrics<3.0,>0.7.0
36
36
  Requires-Dist: tqdm<6.0,>=4.57.0
37
- Requires-Dist: typing-extensions<6.0,>=4.4.0
37
+ Requires-Dist: typing-extensions<6.0,>4.5.0
38
38
  Requires-Dist: pytorch-lightning
39
+ Provides-Extra: pytorch-extra
40
+ Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
41
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-extra"
42
+ Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-extra"
43
+ Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-extra"
44
+ Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
45
+ Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
46
+ Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
39
47
  Provides-Extra: pytorch-strategies
40
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
48
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-strategies"
41
49
  Provides-Extra: pytorch-test
42
50
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-test"
43
- Requires-Dist: coverage==7.9.1; extra == "pytorch-test"
51
+ Requires-Dist: coverage==7.10.5; extra == "pytorch-test"
44
52
  Requires-Dist: fastapi; extra == "pytorch-test"
45
- Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-test"
46
- Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-test"
53
+ Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-test"
54
+ Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-test"
47
55
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-test"
56
+ Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-test"
48
57
  Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-test"
49
58
  Requires-Dist: psutil<8.0; extra == "pytorch-test"
50
59
  Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-test"
51
- Requires-Dist: pytest-random-order==1.1.1; extra == "pytorch-test"
60
+ Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-test"
52
61
  Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-test"
53
62
  Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-test"
54
- Requires-Dist: pytest==8.4.0; extra == "pytorch-test"
63
+ Requires-Dist: pytest==8.4.1; extra == "pytorch-test"
55
64
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-test"
56
- Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "pytorch-test"
65
+ Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-test"
57
66
  Requires-Dist: uvicorn; extra == "pytorch-test"
58
- Provides-Extra: pytorch-extra
59
- Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-extra"
60
- Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-extra"
61
- Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-extra"
62
- Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-extra"
63
- Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-extra"
64
- Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-extra"
65
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-extra"
66
67
  Provides-Extra: pytorch-examples
67
68
  Requires-Dist: ipython[all]<9.0; extra == "pytorch-examples"
68
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-examples"
69
69
  Requires-Dist: requests<3.0; extra == "pytorch-examples"
70
70
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-examples"
71
71
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-examples"
72
72
  Provides-Extra: fabric-strategies
73
73
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-strategies"
74
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
74
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-strategies"
75
75
  Provides-Extra: fabric-test
76
- Requires-Dist: click==8.1.8; extra == "fabric-test"
77
- Requires-Dist: coverage==7.9.1; extra == "fabric-test"
78
- Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-test"
76
+ Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-test"
77
+ Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-test"
78
+ Requires-Dist: coverage==7.10.5; extra == "fabric-test"
79
+ Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-test"
79
80
  Requires-Dist: pytest-cov==6.2.1; extra == "fabric-test"
80
- Requires-Dist: pytest-random-order==1.1.1; extra == "fabric-test"
81
+ Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-test"
81
82
  Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-test"
82
83
  Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-test"
83
- Requires-Dist: pytest==8.4.0; extra == "fabric-test"
84
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-test"
84
+ Requires-Dist: pytest==8.4.1; extra == "fabric-test"
85
+ Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-test"
85
86
  Provides-Extra: fabric-examples
86
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-examples"
87
87
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-examples"
88
88
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-examples"
89
89
  Provides-Extra: fabric-all
90
90
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-all"
91
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
92
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-all"
91
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-all"
93
92
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-all"
94
93
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-all"
95
94
  Provides-Extra: fabric-dev
96
95
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "fabric-dev"
97
- Requires-Dist: click==8.1.8; extra == "fabric-dev"
98
- Requires-Dist: coverage==7.9.1; extra == "fabric-dev"
99
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
100
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "fabric-dev"
101
- Requires-Dist: numpy<2.0,>=1.17.2; extra == "fabric-dev"
96
+ Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "fabric-dev"
97
+ Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "fabric-dev"
98
+ Requires-Dist: coverage==7.10.5; extra == "fabric-dev"
99
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "fabric-dev"
100
+ Requires-Dist: numpy<2.0,>=1.21.0; extra == "fabric-dev"
102
101
  Requires-Dist: pytest-cov==6.2.1; extra == "fabric-dev"
103
- Requires-Dist: pytest-random-order==1.1.1; extra == "fabric-dev"
102
+ Requires-Dist: pytest-random-order==1.2.0; extra == "fabric-dev"
104
103
  Requires-Dist: pytest-rerunfailures==15.1; extra == "fabric-dev"
105
104
  Requires-Dist: pytest-timeout==2.4.0; extra == "fabric-dev"
106
- Requires-Dist: pytest==8.4.0; extra == "fabric-dev"
107
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "fabric-dev"
105
+ Requires-Dist: pytest==8.4.1; extra == "fabric-dev"
106
+ Requires-Dist: tensorboardX<3.0,>=2.6; extra == "fabric-dev"
108
107
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "fabric-dev"
109
108
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "fabric-dev"
110
109
  Provides-Extra: pytorch-all
111
110
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-all"
112
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
111
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-all"
113
112
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-all"
114
113
  Requires-Dist: ipython[all]<9.0; extra == "pytorch-all"
115
114
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-all"
116
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-all"
117
115
  Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-all"
118
116
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-all"
119
117
  Requires-Dist: requests<3.0; extra == "pytorch-all"
@@ -124,66 +122,68 @@ Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-all"
124
122
  Provides-Extra: pytorch-dev
125
123
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "pytorch-dev"
126
124
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "pytorch-dev"
127
- Requires-Dist: coverage==7.9.1; extra == "pytorch-dev"
128
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
125
+ Requires-Dist: coverage==7.10.5; extra == "pytorch-dev"
126
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "pytorch-dev"
129
127
  Requires-Dist: fastapi; extra == "pytorch-dev"
130
128
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "pytorch-dev"
131
129
  Requires-Dist: ipython[all]<9.0; extra == "pytorch-dev"
132
130
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "pytorch-dev"
133
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "pytorch-dev"
134
131
  Requires-Dist: matplotlib<4.0,>3.1; extra == "pytorch-dev"
135
- Requires-Dist: numpy<2.0,>=1.17.2; extra == "pytorch-dev"
132
+ Requires-Dist: numpy<2.0,>1.20.0; extra == "pytorch-dev"
136
133
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "pytorch-dev"
137
- Requires-Dist: onnx<2.0,>=1.12.0; extra == "pytorch-dev"
134
+ Requires-Dist: onnx<2.0,>1.12.0; extra == "pytorch-dev"
138
135
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "pytorch-dev"
136
+ Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "pytorch-dev"
139
137
  Requires-Dist: pandas<3.0,>2.0; extra == "pytorch-dev"
140
138
  Requires-Dist: psutil<8.0; extra == "pytorch-dev"
141
139
  Requires-Dist: pytest-cov==6.2.1; extra == "pytorch-dev"
142
- Requires-Dist: pytest-random-order==1.1.1; extra == "pytorch-dev"
140
+ Requires-Dist: pytest-random-order==1.2.0; extra == "pytorch-dev"
143
141
  Requires-Dist: pytest-rerunfailures==15.1; extra == "pytorch-dev"
144
142
  Requires-Dist: pytest-timeout==2.4.0; extra == "pytorch-dev"
145
- Requires-Dist: pytest==8.4.0; extra == "pytorch-dev"
143
+ Requires-Dist: pytest==8.4.1; extra == "pytorch-dev"
146
144
  Requires-Dist: requests<3.0; extra == "pytorch-dev"
147
145
  Requires-Dist: rich<15.0,>=12.3.0; extra == "pytorch-dev"
148
146
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "pytorch-dev"
149
- Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "pytorch-dev"
147
+ Requires-Dist: tensorboard<3.0,>=2.11; extra == "pytorch-dev"
150
148
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "pytorch-dev"
151
149
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "pytorch-dev"
152
150
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "pytorch-dev"
153
151
  Requires-Dist: uvicorn; extra == "pytorch-dev"
152
+ Provides-Extra: extra
153
+ Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
154
+ Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
155
+ Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
156
+ Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
157
+ Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
158
+ Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
159
+ Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
154
160
  Provides-Extra: strategies
155
161
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "strategies"
156
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
162
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "strategies"
157
163
  Provides-Extra: test
158
- Requires-Dist: click==8.1.8; extra == "test"
164
+ Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "test"
165
+ Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "test"
159
166
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "test"
160
- Requires-Dist: coverage==7.9.1; extra == "test"
167
+ Requires-Dist: coverage==7.10.5; extra == "test"
161
168
  Requires-Dist: fastapi; extra == "test"
162
- Requires-Dist: numpy<2.0,>=1.17.2; extra == "test"
163
- Requires-Dist: onnx<2.0,>=1.12.0; extra == "test"
169
+ Requires-Dist: numpy<2.0,>1.20.0; extra == "test"
170
+ Requires-Dist: numpy<2.0,>=1.21.0; extra == "test"
171
+ Requires-Dist: onnx<2.0,>1.12.0; extra == "test"
164
172
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "test"
173
+ Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "test"
165
174
  Requires-Dist: pandas<3.0,>2.0; extra == "test"
166
175
  Requires-Dist: psutil<8.0; extra == "test"
167
176
  Requires-Dist: pytest-cov==6.2.1; extra == "test"
168
- Requires-Dist: pytest-random-order==1.1.1; extra == "test"
177
+ Requires-Dist: pytest-random-order==1.2.0; extra == "test"
169
178
  Requires-Dist: pytest-rerunfailures==15.1; extra == "test"
170
179
  Requires-Dist: pytest-timeout==2.4.0; extra == "test"
171
- Requires-Dist: pytest==8.4.0; extra == "test"
180
+ Requires-Dist: pytest==8.4.1; extra == "test"
172
181
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "test"
173
- Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "test"
174
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "test"
182
+ Requires-Dist: tensorboard<3.0,>=2.11; extra == "test"
183
+ Requires-Dist: tensorboardX<3.0,>=2.6; extra == "test"
175
184
  Requires-Dist: uvicorn; extra == "test"
176
- Provides-Extra: extra
177
- Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "extra"
178
- Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "extra"
179
- Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "extra"
180
- Requires-Dist: matplotlib<4.0,>3.1; extra == "extra"
181
- Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "extra"
182
- Requires-Dist: rich<15.0,>=12.3.0; extra == "extra"
183
- Requires-Dist: tensorboardX<3.0,>=2.2; extra == "extra"
184
185
  Provides-Extra: examples
185
186
  Requires-Dist: ipython[all]<9.0; extra == "examples"
186
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "examples"
187
187
  Requires-Dist: requests<3.0; extra == "examples"
188
188
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "examples"
189
189
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "examples"
@@ -191,11 +191,10 @@ Provides-Extra: data
191
191
  Requires-Dist: litdata<1.0,>=0.2.0rc; extra == "data"
192
192
  Provides-Extra: all
193
193
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "all"
194
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
194
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "all"
195
195
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "all"
196
196
  Requires-Dist: ipython[all]<9.0; extra == "all"
197
197
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "all"
198
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "all"
199
198
  Requires-Dist: matplotlib<4.0,>3.1; extra == "all"
200
199
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "all"
201
200
  Requires-Dist: requests<3.0; extra == "all"
@@ -205,32 +204,35 @@ Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "all"
205
204
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "all"
206
205
  Provides-Extra: dev
207
206
  Requires-Dist: bitsandbytes<1.0,>=0.45.2; platform_system != "Darwin" and extra == "dev"
208
- Requires-Dist: click==8.1.8; extra == "dev"
207
+ Requires-Dist: click==8.1.8; python_version < "3.11" and extra == "dev"
208
+ Requires-Dist: click==8.2.1; python_version > "3.10" and extra == "dev"
209
209
  Requires-Dist: cloudpickle<4.0,>=1.3; extra == "dev"
210
- Requires-Dist: coverage==7.9.1; extra == "dev"
211
- Requires-Dist: deepspeed<=0.9.3,>=0.8.2; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
210
+ Requires-Dist: coverage==7.10.5; extra == "dev"
211
+ Requires-Dist: deepspeed<=0.15.0,>=0.14.1; (platform_system != "Windows" and platform_system != "Darwin") and extra == "dev"
212
212
  Requires-Dist: fastapi; extra == "dev"
213
213
  Requires-Dist: hydra-core<2.0,>=1.2.0; extra == "dev"
214
214
  Requires-Dist: ipython[all]<9.0; extra == "dev"
215
215
  Requires-Dist: jsonargparse[jsonnet,signatures]<5.0,>=4.39.0; extra == "dev"
216
- Requires-Dist: lightning-utilities<1.0,>=0.8.0; extra == "dev"
217
216
  Requires-Dist: matplotlib<4.0,>3.1; extra == "dev"
218
- Requires-Dist: numpy<2.0,>=1.17.2; extra == "dev"
217
+ Requires-Dist: numpy<2.0,>1.20.0; extra == "dev"
218
+ Requires-Dist: numpy<2.0,>=1.21.0; extra == "dev"
219
219
  Requires-Dist: omegaconf<3.0,>=2.2.3; extra == "dev"
220
- Requires-Dist: onnx<2.0,>=1.12.0; extra == "dev"
220
+ Requires-Dist: onnx<2.0,>1.12.0; extra == "dev"
221
221
  Requires-Dist: onnxruntime<2.0,>=1.12.0; extra == "dev"
222
+ Requires-Dist: onnxscript<1.0,>=0.1.0; extra == "dev"
222
223
  Requires-Dist: pandas<3.0,>2.0; extra == "dev"
223
224
  Requires-Dist: psutil<8.0; extra == "dev"
224
225
  Requires-Dist: pytest-cov==6.2.1; extra == "dev"
225
- Requires-Dist: pytest-random-order==1.1.1; extra == "dev"
226
+ Requires-Dist: pytest-random-order==1.2.0; extra == "dev"
226
227
  Requires-Dist: pytest-rerunfailures==15.1; extra == "dev"
227
228
  Requires-Dist: pytest-timeout==2.4.0; extra == "dev"
228
- Requires-Dist: pytest==8.4.0; extra == "dev"
229
+ Requires-Dist: pytest==8.4.1; extra == "dev"
229
230
  Requires-Dist: requests<3.0; extra == "dev"
230
231
  Requires-Dist: rich<15.0,>=12.3.0; extra == "dev"
231
232
  Requires-Dist: scikit-learn<2.0,>0.22.1; extra == "dev"
232
- Requires-Dist: tensorboard<3.0,>=2.9.1; extra == "dev"
233
+ Requires-Dist: tensorboard<3.0,>=2.11; extra == "dev"
233
234
  Requires-Dist: tensorboardX<3.0,>=2.2; extra == "dev"
235
+ Requires-Dist: tensorboardX<3.0,>=2.6; extra == "dev"
234
236
  Requires-Dist: torchmetrics<2.0,>=0.10.0; extra == "dev"
235
237
  Requires-Dist: torchvision<1.0,>=0.16.0; extra == "dev"
236
238
  Requires-Dist: uvicorn; extra == "dev"
@@ -270,7 +272,7 @@ ______________________________________________________________________
270
272
  <a href="#lightning-fabric-expert-control">Fabric</a> •
271
273
  <a href="https://lightning.ai/">Lightning AI</a> •
272
274
  <a href="#community">Community</a> •
273
- <a href="https://pytorch-lightning.readthedocs.io/en/2.5.2">Docs</a>
275
+ <a href="https://pytorch-lightning.readthedocs.io/en/2.5.4">Docs</a>
274
276
  </p>
275
277
 
276
278
  <!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
@@ -307,6 +309,12 @@ ______________________________________________________________________
307
309
 
308
310
  &nbsp;
309
311
 
312
+ # Why PyTorch Lightning?
313
+
314
+ 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 more DIY.
315
+
316
+ Fun analogy: If PyTorch is Javascript, PyTorch Lightning is ReactJS or NextJS.
317
+
310
318
  # Lightning has 2 core packages
311
319
 
312
320
  [PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
@@ -795,7 +803,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
795
803
 
796
804
  | System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
797
805
  | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
798
- | 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.2)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
806
+ | 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) |
799
807
  | 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) |
800
808
  | 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) |
801
809
  | 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) |
@@ -55,6 +55,12 @@ ______________________________________________________________________
55
55
 
56
56
  &nbsp;
57
57
 
58
+ # Why PyTorch Lightning?
59
+
60
+ 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 more DIY.
61
+
62
+ Fun analogy: If PyTorch is Javascript, PyTorch Lightning is ReactJS or NextJS.
63
+
58
64
  # Lightning has 2 core packages
59
65
 
60
66
  [PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
@@ -179,7 +179,9 @@ markers = [
179
179
  "cloud: Run the cloud tests for example",
180
180
  ]
181
181
  filterwarnings = [
182
+ # "error::DeprecationWarning",
182
183
  "error::FutureWarning",
184
+ "ignore::FutureWarning:onnxscript", # Temporary ignore until onnxscript is updated
183
185
  ]
184
186
  xfail_strict = true
185
187
  junit_duration_report = "call"
@@ -1,9 +1,9 @@
1
- PyYAML<7.0,>=5.4
1
+ PyYAML<7.0,>5.4
2
2
  fsspec[http]<2026.0,>=2022.5.0
3
3
  lightning-utilities<1.0,>=0.10.0
4
4
  packaging<26.0,>=20.0
5
5
  torch<3.0,>=2.1.0
6
- torchmetrics<2.0,>=0.7.0
6
+ torchmetrics<2.0,>0.7.0
7
7
  tqdm<5.0,>=4.57.0
8
- typing-extensions<5.0,>=4.4.0
8
+ typing-extensions<5.0,>4.5.0
9
9
 
@@ -1,8 +1,8 @@
1
1
  # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
- torch >=2.1.0, <2.8.0
5
- fsspec[http] >=2022.5.0, <2025.6.0
4
+ torch >=2.1.0, <2.9.0
5
+ fsspec[http] >=2022.5.0, <2025.8.0
6
6
  packaging >=20.0, <=25.0
7
- typing-extensions >=4.4.0, <4.15.0
8
- lightning-utilities >=0.10.0, <0.15.0
7
+ typing-extensions >4.5.0, <4.15.0
8
+ lightning-utilities >=0.10.0, <0.16.0
@@ -1,6 +1,5 @@
1
1
  # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
- torchvision >=0.16.0, <0.23.0
5
- torchmetrics >=0.10.0, <1.8.0
6
- lightning-utilities >=0.8.0, <0.15.0
4
+ torchvision >=0.16.0, <0.24.0
5
+ torchmetrics >=0.10.0, <1.9.0
@@ -5,5 +5,5 @@
5
5
 
6
6
  # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
7
7
  # shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
8
- deepspeed >=0.8.2, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
8
+ deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin" # strict
9
9
  bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
@@ -0,0 +1,10 @@
1
+ coverage ==7.10.5
2
+ numpy >=1.21.0, <1.27.0
3
+ pytest ==8.4.1
4
+ pytest-cov ==6.2.1
5
+ pytest-timeout ==2.4.0
6
+ pytest-rerunfailures ==15.1
7
+ pytest-random-order ==1.2.0
8
+ click ==8.1.8; python_version < "3.11"
9
+ click ==8.2.1; python_version > "3.10"
10
+ tensorboardX >=2.6, <2.7.0 # todo: relax it back to `>=2.2` after fixing tests
@@ -1,11 +1,11 @@
1
1
  # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
- torch >=2.1.0, <2.8.0
4
+ torch >=2.1.0, <2.9.0
5
5
  tqdm >=4.57.0, <4.68.0
6
- PyYAML >=5.4, <6.1.0
7
- fsspec[http] >=2022.5.0, <2025.6.0
8
- torchmetrics >=0.7.0, <1.8.0
6
+ PyYAML >5.4, <6.1.0
7
+ fsspec[http] >=2022.5.0, <2025.8.0
8
+ torchmetrics >0.7.0, <1.9.0
9
9
  packaging >=20.0, <=25.0
10
- typing-extensions >=4.4.0, <4.15.0
11
- lightning-utilities >=0.10.0, <0.15.0
10
+ typing-extensions >4.5.0, <4.15.0
11
+ lightning-utilities >=0.10.0, <0.16.0
@@ -1,7 +1,9 @@
1
1
  -r ../docs.txt
2
2
 
3
3
  nbformat # used for generate empty notebook
4
- ipython[notebook] <8.19.0
4
+ ipython[notebook] <9.5.0
5
5
  setuptools<81.0 # workaround for `error in ipython setup command: use_2to3 is invalid.`
6
6
 
7
- -r ../../_notebooks/.actions/requires.txt
7
+ onnxscript >= 0.2.2, < 0.5.0
8
+
9
+ #-r ../../_notebooks/.actions/requires.txt
@@ -2,7 +2,6 @@
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
4
  requests <2.33.0
5
- torchvision >=0.16.0, <0.23.0
5
+ torchvision >=0.16.0, <0.24.0
6
6
  ipython[all] <8.19.0
7
- torchmetrics >=0.10.0, <1.8.0
8
- lightning-utilities >=0.8.0, <0.15.0
7
+ torchmetrics >=0.10.0, <1.9.0
@@ -2,10 +2,10 @@
2
2
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
3
 
4
4
  # extended list of package dependencies to reach full functionality
5
- matplotlib>3.1, <3.10.0
5
+ matplotlib>3.1, <3.11.0
6
6
  omegaconf >=2.2.3, <2.4.0
7
7
  hydra-core >=1.2.0, <1.4.0
8
8
  jsonargparse[signatures,jsonnet] >=4.39.0, <4.41.0
9
- rich >=12.3.0, <14.1.0
9
+ rich >=12.3.0, <14.2.0
10
10
  tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute
11
11
  bitsandbytes >=0.45.2,<0.47.0; platform_system != "Darwin"
@@ -3,4 +3,4 @@
3
3
 
4
4
  # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
5
5
  # shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
6
- deepspeed >=0.8.2, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
6
+ deepspeed >=0.14.1,<=0.15.0; platform_system != "Windows" and platform_system != "Darwin" # strict
@@ -1,19 +1,20 @@
1
- coverage ==7.9.1
2
- pytest ==8.4.0
1
+ coverage ==7.10.5
2
+ pytest ==8.4.1
3
3
  pytest-cov ==6.2.1
4
4
  pytest-timeout ==2.4.0
5
5
  pytest-rerunfailures ==15.1
6
- pytest-random-order ==1.1.1
6
+ pytest-random-order ==1.2.0
7
7
 
8
8
  # needed in tests
9
9
  cloudpickle >=1.3, <3.2.0
10
- scikit-learn >0.22.1, <1.7.0
11
- numpy >=1.17.2, <1.27.0
12
- onnx >=1.12.0, <1.19.0
13
- onnxruntime >=1.12.0, <1.21.0
10
+ scikit-learn >0.22.1, <1.8.0
11
+ numpy >1.20.0, <1.27.0
12
+ onnx >1.12.0, <1.19.0
13
+ onnxruntime >=1.12.0, <1.23.0
14
+ onnxscript >= 0.1.0, < 0.5.0
14
15
  psutil <7.0.1 # for `DeviceStatsMonitor`
15
16
  pandas >2.0, <2.4.0 # needed in benchmarks
16
17
  fastapi # for `ServableModuleValidator` # not setting version as re-defined in App
17
18
  uvicorn # for `ServableModuleValidator` # not setting version as re-defined in App
18
19
 
19
- tensorboard >=2.9.1, <2.20.0 # for `TensorBoardLogger`
20
+ tensorboard >=2.11, <2.21.0 # for `TensorBoardLogger`
@@ -15,7 +15,7 @@
15
15
  import time
16
16
 
17
17
  __author__ = "Lightning AI et al."
18
- __author_email__ = "pytorch@lightning.ai"
18
+ __author_email__ = "developer@lightning.ai"
19
19
  __license__ = "Apache-2.0"
20
20
  __copyright__ = f"Copyright (c) 2018-{time.strftime('%Y')}, {__author__}."
21
21
  __homepage__ = "https://github.com/Lightning-AI/lightning"
@@ -6,6 +6,26 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
6
6
 
7
7
  ---
8
8
 
9
+ ## [2.5.4] - 2025-08-29
10
+
11
+ ### Changed
12
+
13
+ - Added support for NVIDIA H200 GPUs in `get_available_flops` ([#20913](https://github.com/Lightning-AI/pytorch-lightning/pull/21119))
14
+
15
+
16
+ ## [2.5.3] - 2025-08-13
17
+
18
+ ### Changed
19
+
20
+ - Enable "auto" for `devices` and `accelerator` as CLI arguments ([#20913](https://github.com/Lightning-AI/pytorch-lightning/pull/20913))
21
+ - Raise ValueError when seed is `out-of-bounds` or `cannot be cast to int` ([#21029](https://github.com/Lightning-AI/pytorch-lightning/pull/21029))
22
+
23
+ ### Fixed
24
+
25
+ - Fixed XLA strategy to add support for `global_ordinal`, `local_ordinal`, `world_size` which came instead of deprecated methods ([#20852](https://github.com/Lightning-AI/pytorch-lightning/issues/20852))
26
+ - Fixed remove extra `name` parameter in accelerator registry decorator ([#20975](https://github.com/Lightning-AI/pytorch-lightning/pull/20975))
27
+
28
+
9
29
  ## [2.5.2] - 2025-3-20
10
30
 
11
31
  ### Changed
@@ -73,14 +73,14 @@ class _AcceleratorRegistry(dict):
73
73
  data["description"] = description
74
74
  data["init_params"] = init_params
75
75
 
76
- def do_register(name: str, accelerator: Callable) -> Callable:
76
+ def do_register(accelerator: Callable) -> Callable:
77
77
  data["accelerator"] = accelerator
78
78
  data["accelerator_name"] = name
79
79
  self[name] = data
80
80
  return accelerator
81
81
 
82
82
  if accelerator is not None:
83
- return do_register(name, accelerator)
83
+ return do_register(accelerator)
84
84
 
85
85
  return do_register
86
86