lightning 2.1.4__tar.gz → 2.2.0.post0__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 (605) hide show
  1. {lightning-2.1.4 → lightning-2.2.0.post0}/.actions/assistant.py +20 -1
  2. {lightning-2.1.4/src/lightning.egg-info → lightning-2.2.0.post0}/PKG-INFO +21 -21
  3. {lightning-2.1.4 → lightning-2.2.0.post0}/README.md +7 -7
  4. {lightning-2.1.4 → lightning-2.2.0.post0}/pyproject.toml +0 -3
  5. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/app/app.txt +1 -1
  6. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/base.txt +2 -2
  7. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/data/test.txt +2 -0
  8. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/fabric/base.txt +2 -2
  9. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/fabric/examples.txt +1 -1
  10. {lightning-2.1.4/requirements/pytorch → lightning-2.2.0.post0/requirements/fabric}/strategies.txt +3 -0
  11. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/pytorch/base.txt +2 -2
  12. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/pytorch/examples.txt +2 -1
  13. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/pytorch/extra.txt +1 -1
  14. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/__setup__.py +3 -1
  15. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/CHANGELOG.md +7 -1
  16. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/README.md +4 -4
  17. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/cmd_init.py +1 -1
  18. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/cmd_react_ui_init.py +3 -3
  19. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/app_commands.py +2 -2
  20. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/ls.py +1 -1
  21. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/README.md +2 -2
  22. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/setup.py +1 -1
  23. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/connect/app.py +7 -7
  24. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/lightning_cli.py +1 -1
  25. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/README.md +2 -2
  26. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/core/flow.py +2 -2
  27. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/cloud.py +1 -1
  28. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/testing/testing.py +1 -1
  29. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/load_app.py +1 -1
  30. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/login.py +1 -1
  31. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/__init__.py +1 -1
  32. {lightning-2.1.4/src/lightning/data/streaming → lightning-2.2.0.post0/src/lightning/data}/constants.py +1 -0
  33. {lightning-2.1.4/src/lightning/data/streaming → lightning-2.2.0.post0/src/lightning/data/processing}/data_processor.py +68 -58
  34. lightning-2.2.0.post0/src/lightning/data/processing/dns.py +47 -0
  35. {lightning-2.1.4/src/lightning/data/streaming → lightning-2.2.0.post0/src/lightning/data/processing}/functions.py +64 -28
  36. lightning-2.2.0.post0/src/lightning/data/processing/image.py +47 -0
  37. lightning-2.2.0.post0/src/lightning/data/processing/readers.py +131 -0
  38. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/__init__.py +0 -4
  39. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/cache.py +4 -2
  40. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/client.py +1 -1
  41. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/config.py +1 -1
  42. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/dataloader.py +1 -1
  43. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/dataset.py +2 -2
  44. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/downloader.py +1 -1
  45. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/item_loader.py +1 -1
  46. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/reader.py +1 -1
  47. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/resolver.py +6 -2
  48. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/serializers.py +3 -3
  49. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/shuffle.py +1 -73
  50. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/writer.py +4 -2
  51. lightning-2.2.0.post0/src/lightning/data/utilities/shuffle.py +79 -0
  52. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/CHANGELOG.md +30 -0
  53. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/accelerators/cpu.py +8 -0
  54. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/accelerators/cuda.py +15 -4
  55. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/accelerators/mps.py +8 -0
  56. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/accelerators/registry.py +3 -0
  57. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/accelerators/xla.py +18 -17
  58. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/cli.py +29 -1
  59. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/connector.py +7 -2
  60. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/fabric.py +33 -10
  61. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/loggers/csv_logs.py +10 -0
  62. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/loggers/tensorboard.py +10 -0
  63. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/collectives/single_device.py +20 -0
  64. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/collectives/torch_collective.py +26 -5
  65. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/kubeflow.py +12 -0
  66. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/lightning.py +16 -1
  67. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/lsf.py +12 -0
  68. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/mpi.py +11 -0
  69. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/slurm.py +13 -0
  70. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/torchelastic.py +12 -0
  71. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/xla.py +14 -2
  72. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/io/torch_io.py +5 -0
  73. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/io/xla.py +2 -0
  74. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/amp.py +9 -0
  75. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/bitsandbytes.py +175 -26
  76. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/deepspeed.py +8 -1
  77. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/double.py +7 -0
  78. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/fsdp.py +11 -1
  79. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/half.py +7 -0
  80. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/transformer_engine.py +31 -11
  81. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/xla.py +3 -1
  82. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/ddp.py +15 -1
  83. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/deepspeed.py +25 -1
  84. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/dp.py +13 -0
  85. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/fsdp.py +49 -31
  86. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/launchers/multiprocessing.py +3 -0
  87. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/launchers/subprocess_script.py +4 -0
  88. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/launchers/xla.py +7 -6
  89. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/parallel.py +5 -0
  90. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/registry.py +3 -0
  91. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/single_device.py +8 -0
  92. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/single_xla.py +6 -0
  93. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/strategy.py +2 -3
  94. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/xla.py +23 -2
  95. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/xla_fsdp.py +33 -5
  96. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/__init__.py +23 -6
  97. lightning-2.2.0.post0/src/lightning/fabric/utilities/consolidate_checkpoint.py +79 -0
  98. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/data.py +47 -0
  99. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/device_dtype_mixin.py +8 -1
  100. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/distributed.py +3 -0
  101. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/imports.py +2 -1
  102. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/init.py +27 -7
  103. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/load.py +88 -4
  104. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/logger.py +5 -1
  105. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/rank_zero.py +33 -8
  106. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/seed.py +7 -12
  107. lightning-2.2.0.post0/src/lightning/fabric/utilities/throughput.py +673 -0
  108. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/types.py +4 -8
  109. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/wrappers.py +71 -8
  110. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/CHANGELOG.md +41 -0
  111. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/_graveyard/precision.py +16 -4
  112. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/accelerators/cpu.py +9 -0
  113. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/accelerators/cuda.py +10 -0
  114. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/accelerators/mps.py +9 -0
  115. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/accelerators/xla.py +4 -0
  116. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/__init__.py +2 -0
  117. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/batch_size_finder.py +12 -4
  118. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/device_stats_monitor.py +9 -0
  119. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/early_stopping.py +9 -7
  120. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/finetuning.py +11 -0
  121. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +4 -0
  122. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/lr_finder.py +3 -0
  123. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/lr_monitor.py +4 -0
  124. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/model_checkpoint.py +19 -6
  125. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/model_summary.py +3 -0
  126. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +4 -0
  127. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/prediction_writer.py +5 -0
  128. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/progress/progress_bar.py +3 -0
  129. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/progress/rich_progress.py +24 -0
  130. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +27 -0
  131. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/pruning.py +7 -2
  132. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/rich_model_summary.py +3 -0
  133. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +8 -0
  134. lightning-2.2.0.post0/src/lightning/pytorch/callbacks/throughput_monitor.py +262 -0
  135. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/timer.py +13 -0
  136. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/cli.py +2 -0
  137. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/hooks.py +38 -8
  138. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/mixins/hparams_mixin.py +2 -1
  139. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/module.py +41 -28
  140. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/optimizer.py +7 -0
  141. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/saving.py +18 -7
  142. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/demos/transformer.py +22 -13
  143. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/comet.py +8 -0
  144. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/csv_logs.py +8 -0
  145. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/logger.py +5 -0
  146. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/mlflow.py +8 -0
  147. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/neptune.py +10 -4
  148. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/tensorboard.py +10 -0
  149. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/wandb.py +60 -0
  150. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/evaluation_loop.py +8 -3
  151. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/fetchers.py +11 -0
  152. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/fit_loop.py +24 -2
  153. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/optimization/automatic.py +4 -0
  154. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/optimization/manual.py +2 -0
  155. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/prediction_loop.py +11 -1
  156. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/progress.py +16 -0
  157. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/training_epoch_loop.py +4 -0
  158. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/utilities.py +2 -2
  159. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/overrides/distributed.py +3 -1
  160. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/io/async_plugin.py +4 -0
  161. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/io/wrapper.py +5 -0
  162. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/layer_sync.py +4 -0
  163. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/amp.py +7 -0
  164. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/deepspeed.py +8 -1
  165. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/double.py +7 -0
  166. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/fsdp.py +11 -1
  167. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/half.py +6 -0
  168. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/precision.py +5 -0
  169. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/xla.py +3 -1
  170. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/profilers/advanced.py +6 -0
  171. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/profilers/base.py +4 -0
  172. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/profilers/pytorch.py +7 -3
  173. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/profilers/simple.py +4 -0
  174. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/profilers/xla.py +4 -0
  175. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/serve/servable_module_validator.py +3 -0
  176. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/ddp.py +27 -12
  177. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/deepspeed.py +180 -156
  178. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/fsdp.py +53 -16
  179. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +4 -0
  180. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +4 -0
  181. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/launchers/xla.py +9 -7
  182. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/parallel.py +6 -0
  183. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/single_device.py +9 -4
  184. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/single_xla.py +26 -4
  185. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/strategy.py +19 -10
  186. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/xla.py +43 -28
  187. lightning-2.2.0.post0/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
  188. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +7 -2
  189. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +7 -5
  190. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +11 -3
  191. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +9 -1
  192. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/signal_connector.py +6 -7
  193. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/trainer.py +15 -18
  194. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/tuner/lr_finder.py +7 -0
  195. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/tuner/tuning.py +5 -2
  196. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/__init__.py +28 -8
  197. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/_pytree.py +1 -1
  198. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/combined_loader.py +35 -1
  199. lightning-2.2.0.post0/src/lightning/pytorch/utilities/consolidate_checkpoint.py +30 -0
  200. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/imports.py +2 -1
  201. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/migration/utils.py +2 -0
  202. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/model_helpers.py +26 -1
  203. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/model_summary/model_summary.py +4 -2
  204. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +7 -0
  205. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/parsing.py +2 -18
  206. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/types.py +1 -0
  207. lightning-2.2.0.post0/src/lightning/version.info +1 -0
  208. {lightning-2.1.4 → lightning-2.2.0.post0/src/lightning.egg-info}/PKG-INFO +21 -21
  209. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning.egg-info/SOURCES.txt +13 -4
  210. lightning-2.2.0.post0/src/lightning.egg-info/entry_points.txt +5 -0
  211. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning.egg-info/requires.txt +41 -25
  212. lightning-2.2.0.post0/src/version.info +1 -0
  213. lightning-2.1.4/src/lightning/version.info +0 -1
  214. lightning-2.1.4/src/lightning.egg-info/entry_points.txt +0 -3
  215. lightning-2.1.4/src/version.info +0 -1
  216. {lightning-2.1.4 → lightning-2.2.0.post0}/CITATION.cff +0 -0
  217. {lightning-2.1.4 → lightning-2.2.0.post0}/LICENSE +0 -0
  218. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/app/cloud.txt +0 -0
  219. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/app/components.txt +0 -0
  220. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/app/docs.txt +0 -0
  221. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/app/test.txt +0 -0
  222. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/app/ui.txt +0 -0
  223. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/data/cloud.txt +0 -0
  224. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/data/data.txt +0 -0
  225. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/data/examples.txt +0 -0
  226. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/fabric/docs.txt +0 -0
  227. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/fabric/test.txt +0 -0
  228. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/pytorch/docs.txt +0 -0
  229. {lightning-2.1.4/requirements/fabric → lightning-2.2.0.post0/requirements/pytorch}/strategies.txt +0 -0
  230. {lightning-2.1.4 → lightning-2.2.0.post0}/requirements/pytorch/test.txt +0 -0
  231. {lightning-2.1.4 → lightning-2.2.0.post0}/setup.cfg +0 -0
  232. {lightning-2.1.4 → lightning-2.2.0.post0}/setup.py +0 -0
  233. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/__about__.py +0 -0
  234. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/__init__.py +0 -0
  235. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/__main__.py +0 -0
  236. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/__version__.py +0 -0
  237. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/__init__.py +0 -0
  238. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/api/__init__.py +0 -0
  239. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/api/http_methods.py +0 -0
  240. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/api/request_types.py +0 -0
  241. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/__init__.py +0 -0
  242. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/.gitignore +0 -0
  243. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/LICENSE +0 -0
  244. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/app.py +0 -0
  245. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/placeholdername/__init__.py +0 -0
  246. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/placeholdername/components/component_a/__init__.py +0 -0
  247. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/placeholdername/components/component_a/component_a.py +0 -0
  248. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/placeholdername/components/component_b/__init__.py +0 -0
  249. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/placeholdername/components/component_b/component_a.py +0 -0
  250. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/requirements.txt +0 -0
  251. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/setup.py +0 -0
  252. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/tests/README.md +0 -0
  253. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/tests/__init__.py +0 -0
  254. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/tests/requirements.txt +0 -0
  255. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/app-template/tests/test_placeholdername_app.py +0 -0
  256. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/cmd_apps.py +0 -0
  257. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/cmd_install.py +0 -0
  258. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/cmd_pl_init.py +0 -0
  259. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/__init__.py +0 -0
  260. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/cd.py +0 -0
  261. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/cp.py +0 -0
  262. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/logs.py +0 -0
  263. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/pwd.py +0 -0
  264. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/commands/rm.py +0 -0
  265. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/.github/workflows/ci-testing.yml +0 -0
  266. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/.gitignore +0 -0
  267. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/LICENSE +0 -0
  268. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/app.py +0 -0
  269. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/placeholdername/__init__.py +0 -0
  270. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/placeholdername/component.py +0 -0
  271. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/requirements.txt +0 -0
  272. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/tests/README.md +0 -0
  273. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/tests/__init__.py +0 -0
  274. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/tests/requirements.txt +0 -0
  275. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/component-template/tests/test_placeholdername_component.py +0 -0
  276. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/connect/__init__.py +0 -0
  277. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/connect/data.py +0 -0
  278. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/core.py +0 -0
  279. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/lightning_cli_delete.py +0 -0
  280. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/lightning_cli_launch.py +0 -0
  281. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/lightning_cli_list.py +0 -0
  282. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/.gitignore +0 -0
  283. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/.lightningignore +0 -0
  284. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/app.py +0 -0
  285. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/__init__.py +0 -0
  286. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/callbacks.py +0 -0
  287. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/components/__init__.py +0 -0
  288. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/components/logger/__init__.py +0 -0
  289. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/components/logger/tensorboard.py +0 -0
  290. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/components/logger/weights_and_biases.py +0 -0
  291. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/components/script_runner/__init__.py +0 -0
  292. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/components/script_runner/script_runner.py +0 -0
  293. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/core/state.py +0 -0
  294. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/setup.py +0 -0
  295. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/tests/__init__.py +0 -0
  296. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/tests/core/__init__.py +0 -0
  297. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/tests/core/test_callbacks.py +0 -0
  298. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/tests/test_app.py +0 -0
  299. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/.gitignore +0 -0
  300. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/.prettierignore +0 -0
  301. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/.prettierrc +0 -0
  302. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/craco.config.js +0 -0
  303. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/package.json +0 -0
  304. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/public/favicon.svg +0 -0
  305. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/public/index.html +0 -0
  306. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/public/manifest.json +0 -0
  307. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/public/robots.txt +0 -0
  308. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/App.tsx +0 -0
  309. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/EnvironmentConfigurator.tsx +0 -0
  310. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/ErrorPanel.tsx +0 -0
  311. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/ExecutionSummary.tsx +0 -0
  312. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/HyperparameterSummary.tsx +0 -0
  313. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/Launcher.tsx +0 -0
  314. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/ProgressBar.tsx +0 -0
  315. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/ProgressBarGroup.tsx +0 -0
  316. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/components/Timer.tsx +0 -0
  317. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/hooks/useLightningState.ts +0 -0
  318. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/index.css +0 -0
  319. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/index.tsx +0 -0
  320. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/lightning-colors.ts +0 -0
  321. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/react-app-env.d.ts +0 -0
  322. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/reportWebVitals.ts +0 -0
  323. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/src/types/lightning.ts +0 -0
  324. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/pl-app-template/ui/tsconfig.json +0 -0
  325. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/example_app.py +0 -0
  326. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/index.html +0 -0
  327. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/package.json +0 -0
  328. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/App.css +0 -0
  329. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/App.tsx +0 -0
  330. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/favicon.svg +0 -0
  331. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/hooks/useLightningState.ts +0 -0
  332. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/index.css +0 -0
  333. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/main.tsx +0 -0
  334. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/types/lightning.ts +0 -0
  335. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/src/vite-env.d.ts +0 -0
  336. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/tsconfig.json +0 -0
  337. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/tsconfig.node.json +0 -0
  338. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/vite.config.ts +0 -0
  339. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/cli/react-ui-template/ui/yarn.lock +0 -0
  340. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/__init__.py +0 -0
  341. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/database/__init__.py +0 -0
  342. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/database/client.py +0 -0
  343. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/database/server.py +0 -0
  344. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/database/utilities.py +0 -0
  345. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/multi_node/__init__.py +0 -0
  346. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/multi_node/base.py +0 -0
  347. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/multi_node/fabric.py +0 -0
  348. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/multi_node/pytorch_spawn.py +0 -0
  349. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/multi_node/trainer.py +0 -0
  350. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/python/__init__.py +0 -0
  351. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/python/popen.py +0 -0
  352. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/python/tracer.py +0 -0
  353. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/__init__.py +0 -0
  354. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/auto_scaler.py +0 -0
  355. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/catimage.png +0 -0
  356. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/cold_start_proxy.py +0 -0
  357. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/gradio_server.py +0 -0
  358. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/python_server.py +0 -0
  359. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/serve.py +0 -0
  360. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/streamlit.py +0 -0
  361. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/types/__init__.py +0 -0
  362. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/types/image.py +0 -0
  363. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/serve/types/type.py +0 -0
  364. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/components/training.py +0 -0
  365. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/core/__init__.py +0 -0
  366. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/core/api.py +0 -0
  367. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/core/app.py +0 -0
  368. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/core/constants.py +0 -0
  369. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/core/queues.py +0 -0
  370. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/core/work.py +0 -0
  371. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/__init__.py +0 -0
  372. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/frontend.py +0 -0
  373. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/just_py/__init__.py +0 -0
  374. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/just_py/just_py.py +0 -0
  375. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/just_py/just_py_base.py +0 -0
  376. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/panel/__init__.py +0 -0
  377. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/panel/app_state_comm.py +0 -0
  378. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/panel/app_state_watcher.py +0 -0
  379. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/panel/panel_frontend.py +0 -0
  380. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/panel/panel_serve_render_fn.py +0 -0
  381. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/stream_lit.py +0 -0
  382. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/streamlit_base.py +0 -0
  383. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/utils.py +0 -0
  384. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/frontend/web.py +0 -0
  385. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/launcher/__init__.py +0 -0
  386. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/launcher/launcher.py +0 -0
  387. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/launcher/lightning_backend.py +0 -0
  388. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/launcher/lightning_hybrid_backend.py +0 -0
  389. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/pdb/__init__.py +0 -0
  390. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/pdb/pdb.py +0 -0
  391. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/plugin/__init__.py +0 -0
  392. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/plugin/plugin.py +0 -0
  393. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/__init__.py +0 -0
  394. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/backends/__init__.py +0 -0
  395. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/backends/backend.py +0 -0
  396. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/backends/cloud.py +0 -0
  397. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/backends/docker.py +0 -0
  398. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/backends/mp_process.py +0 -0
  399. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/multiprocess.py +0 -0
  400. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/runtime.py +0 -0
  401. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/runners/runtime_type.py +0 -0
  402. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/source_code/__init__.py +0 -0
  403. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/source_code/copytree.py +0 -0
  404. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/source_code/hashing.py +0 -0
  405. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/source_code/local.py +0 -0
  406. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/source_code/tar.py +0 -0
  407. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/source_code/uploader.py +0 -0
  408. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/__init__.py +0 -0
  409. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/copier.py +0 -0
  410. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/drive.py +0 -0
  411. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/filesystem.py +0 -0
  412. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/mount.py +0 -0
  413. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/orchestrator.py +0 -0
  414. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/path.py +0 -0
  415. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/payload.py +0 -0
  416. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/storage/requests.py +0 -0
  417. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/structures/__init__.py +0 -0
  418. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/structures/dict.py +0 -0
  419. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/structures/list.py +0 -0
  420. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/testing/__init__.py +0 -0
  421. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/testing/config.py +0 -0
  422. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/testing/helpers.py +0 -0
  423. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/asset-manifest.json +0 -0
  424. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/index.html +0 -0
  425. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/css/main.fb7060be.css +0 -0
  426. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/css/main.fb7060be.css.map +0 -0
  427. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/favicon.ico +0 -0
  428. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/js/787.418637a8.chunk.js +0 -0
  429. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/js/787.418637a8.chunk.js.map +0 -0
  430. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/js/main.c1f02aeb.js +0 -0
  431. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/js/main.c1f02aeb.js.LICENSE.txt +0 -0
  432. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/js/main.c1f02aeb.js.map +0 -0
  433. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/lightningState.js +0 -0
  434. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/manifest.json +0 -0
  435. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/media/error.11892047d0183a4723b91dc0fb98cb95.svg +0 -0
  436. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/media/lightning-logo-with-text.b964c8fbf221c97eb8ce52bb6e3a30d6.svg +0 -0
  437. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/media/success.5edae4c5b171e2c1c5a3c54273c47ba8.svg +0 -0
  438. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/media/warning.5c542673e84e77ceb6a230bfea7f7263.svg +0 -0
  439. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/static/robots.txt +0 -0
  440. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/ui/version.info +0 -0
  441. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/__init__.py +0 -0
  442. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/app_commands.py +0 -0
  443. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/app_helpers.py +0 -0
  444. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/app_logs.py +0 -0
  445. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/app_status.py +0 -0
  446. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/auth.py +0 -0
  447. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/cli_helpers.py +0 -0
  448. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/cloud.py +0 -0
  449. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/clusters.py +0 -0
  450. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/commands/__init__.py +0 -0
  451. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/commands/base.py +0 -0
  452. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/component.py +0 -0
  453. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/data_structures.py +0 -0
  454. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/dependency_caching.py +0 -0
  455. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/enum.py +0 -0
  456. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/exceptions.py +0 -0
  457. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/frontend.py +0 -0
  458. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/git.py +0 -0
  459. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/imports.py +0 -0
  460. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/introspection.py +0 -0
  461. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/layout.py +0 -0
  462. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/log.py +0 -0
  463. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/log_helpers.py +0 -0
  464. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/logs_socket_api.py +0 -0
  465. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/name_generator.py +0 -0
  466. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/network.py +0 -0
  467. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/openapi.py +0 -0
  468. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/packaging/__init__.py +0 -0
  469. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/packaging/app_config.py +0 -0
  470. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/packaging/build_config.py +0 -0
  471. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/packaging/cloud_compute.py +0 -0
  472. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/packaging/docker.py +0 -0
  473. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/packaging/lightning_utils.py +0 -0
  474. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/packaging/tarfile.py +0 -0
  475. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/port.py +0 -0
  476. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/proxies.py +0 -0
  477. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/redis.py +0 -0
  478. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/safe_pickle.py +0 -0
  479. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/scheduler.py +0 -0
  480. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/secrets.py +0 -0
  481. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/state.py +0 -0
  482. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/tracer.py +0 -0
  483. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/tree.py +0 -0
  484. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/types.py +0 -0
  485. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/app/utilities/warnings.py +0 -0
  486. {lightning-2.1.4/src/lightning/data/utilities → lightning-2.2.0.post0/src/lightning/data/processing}/__init__.py +0 -0
  487. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/combined.py +0 -0
  488. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/compression.py +0 -0
  489. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/streaming/sampler.py +0 -0
  490. {lightning-2.1.4/src/lightning/pytorch/overrides → lightning-2.2.0.post0/src/lightning/data/utilities}/__init__.py +0 -0
  491. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/utilities/broadcast.py +0 -0
  492. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/utilities/env.py +0 -0
  493. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/utilities/format.py +0 -0
  494. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/data/utilities/packing.py +0 -0
  495. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/__init__.py +0 -0
  496. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/_graveyard/__init__.py +0 -0
  497. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/_graveyard/tpu.py +0 -0
  498. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/accelerators/__init__.py +0 -0
  499. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/accelerators/accelerator.py +0 -0
  500. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/loggers/__init__.py +0 -0
  501. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/loggers/logger.py +0 -0
  502. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/__init__.py +0 -0
  503. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
  504. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/collectives/collective.py +0 -0
  505. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
  506. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
  507. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/io/__init__.py +0 -0
  508. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/io/checkpoint_io.py +0 -0
  509. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
  510. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/precision.py +0 -0
  511. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/plugins/precision/utils.py +0 -0
  512. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/__init__.py +0 -0
  513. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
  514. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
  515. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/apply_func.py +0 -0
  516. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/cloud_io.py +0 -0
  517. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/device_parser.py +0 -0
  518. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/enums.py +0 -0
  519. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/exceptions.py +0 -0
  520. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/optimizer.py +0 -0
  521. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/registry.py +0 -0
  522. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/spike.py +0 -0
  523. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
  524. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/testing/_runif.py +0 -0
  525. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/fabric/utilities/warnings.py +0 -0
  526. /lightning-2.1.4/src/lightning/72bd00aaa7929e5486d97a935be5252a8d8da97e8154ab66 → /lightning-2.2.0.post0/src/lightning/ffb6a4864f9575f98c1278129cf17f7a864beee42ceef0d4 +0 -0
  527. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/py.typed +0 -0
  528. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/__init__.py +0 -0
  529. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/_graveyard/__init__.py +0 -0
  530. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
  531. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
  532. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/_graveyard/ipu.py +0 -0
  533. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
  534. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/accelerators/__init__.py +0 -0
  535. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/accelerators/accelerator.py +0 -0
  536. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/callback.py +0 -0
  537. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
  538. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
  539. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
  540. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/callbacks/spike.py +0 -0
  541. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/__init__.py +0 -0
  542. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/datamodule.py +0 -0
  543. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
  544. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/demos/__init__.py +0 -0
  545. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/demos/boring_classes.py +0 -0
  546. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/demos/mnist_datamodule.py +0 -0
  547. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/__init__.py +0 -0
  548. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loggers/utilities.py +0 -0
  549. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/__init__.py +0 -0
  550. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/loop.py +0 -0
  551. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
  552. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/loops/optimization/closure.py +0 -0
  553. {lightning-2.1.4/src/lightning/pytorch/trainer/connectors → lightning-2.2.0.post0/src/lightning/pytorch/overrides}/__init__.py +0 -0
  554. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/__init__.py +0 -0
  555. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
  556. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
  557. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
  558. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
  559. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
  560. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
  561. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +0 -0
  562. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
  563. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/profilers/__init__.py +0 -0
  564. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/profilers/profiler.py +0 -0
  565. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/serve/__init__.py +0 -0
  566. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/serve/servable_module.py +0 -0
  567. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/__init__.py +0 -0
  568. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
  569. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
  570. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/__init__.py +0 -0
  571. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/call.py +0 -0
  572. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/configuration_validator.py +0 -0
  573. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/callback_connector.py +0 -0
  574. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/data_connector.py +0 -0
  575. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
  576. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +0 -0
  577. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/setup.py +0 -0
  578. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/trainer/states.py +0 -0
  579. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/tuner/__init__.py +0 -0
  580. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/tuner/batch_size_scaling.py +0 -0
  581. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/argparse.py +0 -0
  582. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/compile.py +0 -0
  583. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/data.py +0 -0
  584. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/deepspeed.py +0 -0
  585. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/enums.py +0 -0
  586. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/exceptions.py +0 -0
  587. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/grads.py +0 -0
  588. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/memory.py +0 -0
  589. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
  590. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/migration/migration.py +0 -0
  591. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
  592. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/parameter_tying.py +0 -0
  593. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
  594. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/seed.py +0 -0
  595. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
  596. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
  597. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/testing/_runif.py +0 -0
  598. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +0 -0
  599. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/pytorch/utilities/warnings.py +0 -0
  600. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/store/__init__.py +0 -0
  601. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/store/store.py +0 -0
  602. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning/store/utils.py +0 -0
  603. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning.egg-info/dependency_links.txt +0 -0
  604. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning.egg-info/not-zip-safe +0 -0
  605. {lightning-2.1.4 → lightning-2.2.0.post0}/src/lightning.egg-info/top_level.txt +0 -0
@@ -442,9 +442,20 @@ class AssistantCLI:
442
442
  target_dir: str = "docs/source-pytorch/XXX",
443
443
  checkout: str = "refs/tags/1.0.0",
444
444
  source_dir: str = "docs/source",
445
+ single_page: Optional[str] = None,
445
446
  as_orphan: bool = False,
446
447
  ) -> None:
447
- """Pull docs pages from external source and append to local docs."""
448
+ """Pull docs pages from external source and append to local docs.
449
+
450
+ Args:
451
+ gh_user_repo: standard GitHub user/repo string
452
+ target_dir: relative location inside the docs folder
453
+ checkout: specific tag or branch to checkout
454
+ source_dir: relative location inside the remote / external repo
455
+ single_page: copy only single page from the remote repo and name it as the repo name
456
+ as_orphan: append orphan statement to the page
457
+
458
+ """
448
459
  import zipfile
449
460
 
450
461
  zip_url = f"https://github.com/{gh_user_repo}/archive/{checkout}.zip"
@@ -464,6 +475,14 @@ class AssistantCLI:
464
475
  assert len(zip_dirs) == 1
465
476
  repo_dir = zip_dirs[0]
466
477
 
478
+ if single_page: # special case for copying single page
479
+ single_page = os.path.join(repo_dir, source_dir, single_page)
480
+ assert os.path.isfile(single_page), f"File '{single_page}' does not exist."
481
+ name = re.sub(r"lightning[-_]?", "", gh_user_repo.split("/")[-1])
482
+ new_rst = os.path.join(_PROJECT_ROOT, target_dir, f"{name}.rst")
483
+ AssistantCLI._copy_rst(single_page, new_rst, as_orphan=as_orphan)
484
+ return
485
+ # continue with copying all pages
467
486
  ls_pages = glob.glob(os.path.join(repo_dir, source_dir, "*.rst"))
468
487
  ls_pages += glob.glob(os.path.join(repo_dir, source_dir, "**", "*.rst"))
469
488
  for rst in ls_pages:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: lightning
3
- Version: 2.1.4
3
+ Version: 2.2.0.post0
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
  Author: Lightning AI et al.
@@ -27,21 +27,21 @@ Classifier: Programming Language :: Python :: 3.10
27
27
  Classifier: Programming Language :: Python :: 3.11
28
28
  Requires-Python: >=3.8
29
29
  Description-Content-Type: text/markdown
30
+ Provides-Extra: data-cloud
31
+ Provides-Extra: data-test
32
+ Provides-Extra: data-examples
33
+ Provides-Extra: pytorch-test
34
+ Provides-Extra: pytorch-examples
35
+ Provides-Extra: pytorch-strategies
36
+ Provides-Extra: pytorch-extra
30
37
  Provides-Extra: store-test
31
- Provides-Extra: app-ui
32
38
  Provides-Extra: app-cloud
33
39
  Provides-Extra: app-test
40
+ Provides-Extra: app-ui
34
41
  Provides-Extra: app-components
35
- Provides-Extra: data-cloud
36
- Provides-Extra: data-test
37
- Provides-Extra: data-examples
38
42
  Provides-Extra: fabric-test
39
43
  Provides-Extra: fabric-examples
40
44
  Provides-Extra: fabric-strategies
41
- Provides-Extra: pytorch-test
42
- Provides-Extra: pytorch-extra
43
- Provides-Extra: pytorch-examples
44
- Provides-Extra: pytorch-strategies
45
45
  Provides-Extra: fabric-all
46
46
  Provides-Extra: fabric-dev
47
47
  Provides-Extra: pytorch-all
@@ -51,15 +51,15 @@ Provides-Extra: app-all
51
51
  Provides-Extra: app-dev
52
52
  Provides-Extra: data-all
53
53
  Provides-Extra: data-dev
54
- Provides-Extra: test
55
- Provides-Extra: ui
56
54
  Provides-Extra: cloud
57
- Provides-Extra: app
58
- Provides-Extra: components
55
+ Provides-Extra: test
59
56
  Provides-Extra: data
60
57
  Provides-Extra: examples
61
58
  Provides-Extra: strategies
62
59
  Provides-Extra: extra
60
+ Provides-Extra: app
61
+ Provides-Extra: ui
62
+ Provides-Extra: components
63
63
  Provides-Extra: all
64
64
  Provides-Extra: dev
65
65
  Provides-Extra: store
@@ -83,7 +83,7 @@ ______________________________________________________________________
83
83
  <a href="https://lightning.ai/docs/pytorch/stable/">PyTorch Lightning</a> •
84
84
  <a href="https://lightning.ai/docs/fabric/stable/">Fabric</a> •
85
85
  <a href="https://lightning.ai/docs/app/stable/">Lightning Apps</a> •
86
- <a href="https://pytorch-lightning.readthedocs.io/en/2.1.4">Docs</a> •
86
+ <a href="https://pytorch-lightning.readthedocs.io/en/2.2.0.post0">Docs</a> •
87
87
  <a href="#community">Community</a> •
88
88
  <a href="https://lightning.ai/docs/pytorch/stable/generated/CONTRIBUTING.html">Contribute</a> •
89
89
  </p>
@@ -624,13 +624,13 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
624
624
 
625
625
  <center>
626
626
 
627
- | System / PyTorch ver. | 1.12 | 1.13 | 2.0 | 2.1 |
628
- | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
629
- | 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.1.4)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
630
- | Linux py3.9 \[TPUs\] | | | [![Test PyTorch - TPU](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml) | |
631
- | 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) | [![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) |
632
- | 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) | [![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) |
633
- | 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) | [![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) |
627
+ | System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
628
+ | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
629
+ | 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.2.0.post0)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
630
+ | Linux py3.9 \[TPUs\] | | [![Test PyTorch - TPU](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml) | |
631
+ | 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) |
632
+ | 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) |
633
+ | 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) |
634
634
 
635
635
  </center>
636
636
  </details>
@@ -598,13 +598,13 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
598
598
 
599
599
  <center>
600
600
 
601
- | System / PyTorch ver. | 1.12 | 1.13 | 2.0 | 2.1 |
602
- | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
603
- | Linux py3.9 \[GPUs\] | | | | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=master)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
604
- | Linux py3.9 \[TPUs\] | | | [![Test PyTorch - TPU](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml) | |
605
- | 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) | [![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) |
606
- | 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) | [![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) |
607
- | 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) | [![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) |
601
+ | System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
602
+ | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
603
+ | Linux py3.9 \[GPUs\] | | | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=master)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
604
+ | Linux py3.9 \[TPUs\] | | [![Test PyTorch - TPU](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml) | |
605
+ | 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) |
606
+ | 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) |
607
+ | 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) |
608
608
 
609
609
  </center>
610
610
  </details>
@@ -79,7 +79,6 @@ ignore-init-module-imports = true
79
79
  ".actions/*" = ["S101", "S310"]
80
80
  "setup.py" = ["S101"]
81
81
  "examples/**" = [
82
- "F841", # Local variable is assigned to but never used
83
82
  "S101", # Use of `assert` detected
84
83
  "S113", # todo: Probable use of requests call without
85
84
  "S104", # Possible binding to all interface
@@ -89,7 +88,6 @@ ignore-init-module-imports = true
89
88
  "S108", # Probable insecure usage of temporary file or directory: "/tmp/data/MNIST"
90
89
  ]
91
90
  "src/**" = [
92
- "F841", # Local variable is assigned to but never used
93
91
  "S101", # todo: Use of `assert` detected
94
92
  "S105", "S106", "S107", # todo: Possible hardcoded password: ...
95
93
  "S113", # todo: Probable use of requests call without timeout
@@ -105,7 +103,6 @@ ignore-init-module-imports = true
105
103
  "RET503",
106
104
  ]
107
105
  "tests/**" = [
108
- "F841", # Local variable is assigned to but never used
109
106
  "S101", # Use of `assert` detected
110
107
  "S105", "S106", # todo: Possible hardcoded password: ...
111
108
  "S301", # `pickle` and modules that wrap it can be unsafe when used to deserialize untrusted data, possible security issue
@@ -1,6 +1,6 @@
1
1
  lightning-cloud == 0.5.64 # Must be pinned to ensure compatibility
2
2
  packaging
3
- typing-extensions >=4.0.0, <4.8.0
3
+ typing-extensions >=4.4.0, <4.10.0
4
4
  deepdiff >=5.7.0, <6.6.0
5
5
  fsspec[http] >=2022.5.0, <2023.11.0
6
6
  croniter >=1.3.0, <1.5.0 # strict; TODO: for now until we find something more robust.
@@ -3,8 +3,8 @@ fsspec[http]<2024.0,>=2022.5.0
3
3
  lightning-utilities<1.0,>=0.8.0
4
4
  numpy<2.0,>=1.17.2
5
5
  packaging<24.0,>=20.0
6
- torch<3.0,>=1.12.0
6
+ torch<3.0,>=1.13.0
7
7
  torchmetrics<2.0,>=0.7.0
8
8
  tqdm<5.0,>=4.57.0
9
- typing-extensions<5.0,>=4.0.0
9
+ typing-extensions<5.0,>=4.4.0
10
10
 
@@ -5,3 +5,5 @@ pytest-timeout ==2.1.0
5
5
  pytest-rerunfailures ==12.0
6
6
  pytest-random-order ==1.1.0
7
7
  viztracer
8
+ pyarrow
9
+ polars
@@ -2,8 +2,8 @@
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
  numpy >=1.17.2, <1.27.0
5
- torch >=1.12.0, <2.2.0
5
+ torch >=1.13.0, <2.2.0
6
6
  fsspec[http] >=2022.5.0, <2023.11.0
7
7
  packaging >=20.0, <=23.1
8
- typing-extensions >=4.0.0, <4.8.0
8
+ typing-extensions >=4.4.0, <4.10.0
9
9
  lightning-utilities >=0.8.0, <0.10.0
@@ -1,6 +1,6 @@
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.13.0, <0.17.0
4
+ torchvision >=0.14.0, <0.17.0
5
5
  torchmetrics >=0.10.0, <1.3.0
6
6
  lightning-utilities >=0.8.0, <0.10.0
@@ -1,6 +1,9 @@
1
+ # Despite its misleading name, this file contains dependencies for testing that only get installed in the GPU CI
2
+
1
3
  # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
2
4
  # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
3
5
 
4
6
  # note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
5
7
  # shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
6
8
  deepspeed >=0.8.2, <=0.9.3; platform_system != "Windows" # strict
9
+ bitsandbytes ==0.41.0 # strict
@@ -2,11 +2,11 @@
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
  numpy >=1.17.2, <1.27.0
5
- torch >=1.12.0, <2.2.0
5
+ torch >=1.13.0, <2.2.0
6
6
  tqdm >=4.57.0, <4.67.0
7
7
  PyYAML >=5.4, <6.1.0
8
8
  fsspec[http] >=2022.5.0, <2023.11.0
9
9
  torchmetrics >=0.7.0, <1.3.0 # needed for using fixed compare_version
10
10
  packaging >=20.0, <=23.1
11
- typing-extensions >=4.0.0, <4.8.0
11
+ typing-extensions >=4.4.0, <4.10.0
12
12
  lightning-utilities >=0.8.0, <0.10.0
@@ -1,7 +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
- torchvision >=0.13.0, <0.17.0
4
+ requests <2.32.0
5
+ torchvision >=0.14.0, <0.17.0
5
6
  gym[classic_control] >=0.17.0, <0.27.0
6
7
  ipython[all] <8.15.0
7
8
  torchmetrics >=0.10.0, <1.3.0
@@ -8,4 +8,4 @@ hydra-core >=1.0.5, <1.4.0
8
8
  jsonargparse[signatures] >=4.26.1, <4.28.0
9
9
  rich >=12.3.0, <13.6.0
10
10
  tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute
11
- bitsandbytes <=0.41.1
11
+ bitsandbytes ==0.41.0 # strict
@@ -114,7 +114,9 @@ def _setup_args() -> Dict[str, Any]:
114
114
  "python_requires": ">=3.8", # todo: take the lowes based on all packages
115
115
  "entry_points": {
116
116
  "console_scripts": [
117
- "lightning = lightning:_cli_entry_point",
117
+ "fabric = lightning.fabric.cli:_main",
118
+ "lightning = lightning.fabric.cli:_legacy_main",
119
+ "lightning_app = lightning:_cli_entry_point",
118
120
  ],
119
121
  },
120
122
  "setup_requires": [],
@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
4
4
 
5
5
  The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
6
6
 
7
+ ## [2.2.0] - 2024-02-09
8
+
9
+ ## Changed
10
+
11
+ - Renames the `lightning` cli to `lightning_app` ([#19440](https://github.com/Lightning-AI/pytorch-lightning/pull/19440))
12
+
7
13
 
8
14
  ## [2.1.4] - 2024-01-31
9
15
 
@@ -34,7 +40,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
34
40
 
35
41
  ### Added
36
42
 
37
- - add flow `fail()` ([#18883](https://github.com/Lightning-AI/lightning/pull/18883))
43
+ - Added flow `fail()` ([#18883](https://github.com/Lightning-AI/lightning/pull/18883))
38
44
 
39
45
  ### Fixed
40
46
 
@@ -3,7 +3,7 @@
3
3
  This ⚡ [Lightning app](https://lightning.ai/) ⚡ was generated automatically with:
4
4
 
5
5
  ```bash
6
- lightning init app placeholdername
6
+ lightning_app init app placeholdername
7
7
  ```
8
8
 
9
9
  ## To run placeholdername
@@ -11,19 +11,19 @@ lightning init app placeholdername
11
11
  First, install placeholdername (warning: this app has not been officially approved on the lightning gallery):
12
12
 
13
13
  ```bash
14
- lightning install app https://github.com/theUser/placeholdername
14
+ lightning_app install app https://github.com/theUser/placeholdername
15
15
  ```
16
16
 
17
17
  Once the app is installed, run it locally with:
18
18
 
19
19
  ```bash
20
- lightning run app placeholdername/app.py
20
+ lightning_app run app placeholdername/app.py
21
21
  ```
22
22
 
23
23
  Run it on the [lightning cloud](https://lightning.ai/) with:
24
24
 
25
25
  ```bash
26
- lightning run app placeholdername/app.py --cloud
26
+ lightning_app run app placeholdername/app.py --cloud
27
27
  ```
28
28
 
29
29
  ## to test and link
@@ -27,7 +27,7 @@ def app(app_name: str) -> None:
27
27
  app_name = _capture_valid_app_component_name(resource_type="app")
28
28
 
29
29
  # generate resource template
30
- new_resource_name, name_for_files = _make_resource(resource_dir="app-template", resource_name=app_name)
30
+ new_resource_name, _ = _make_resource(resource_dir="app-template", resource_name=app_name)
31
31
 
32
32
  m = f"""
33
33
  ⚡ Lightning app template created! ⚡
@@ -59,7 +59,7 @@ def _copy_and_setup_react_ui(dest_dir: Optional[str] = None) -> None:
59
59
  return la.frontend.StaticWebFrontend(Path(__file__).parent / "react-ui/src/dist")
60
60
 
61
61
  ⚡ run the example_app.py to see it live!
62
- lightning run app {dest_dir}/example_app.py
62
+ lightning_app run app {dest_dir}/example_app.py
63
63
 
64
64
  """
65
65
  logger.info(m)
@@ -80,7 +80,7 @@ def _check_react_prerequisites() -> None:
80
80
 
81
81
  if not has_npm:
82
82
  m = """
83
- This machine is missing 'npm'. Please install npm and rerun 'lightning init react-ui' again.
83
+ This machine is missing 'npm'. Please install npm and rerun 'lightning_app init react-ui' again.
84
84
 
85
85
  Install instructions: https://docs.npmjs.com/downloading-and-installing-node-js-and-npm
86
86
  """
@@ -94,7 +94,7 @@ def _check_react_prerequisites() -> None:
94
94
 
95
95
  if not has_node:
96
96
  m = """
97
- This machine is missing 'node'. Please install node and rerun 'lightning init react-ui' again.
97
+ This machine is missing 'node'. Please install node and rerun 'lightning_app init react-ui' again.
98
98
 
99
99
  Install instructions: https://docs.npmjs.com/downloading-and-installing-node-js-and-npm
100
100
  """
@@ -61,7 +61,7 @@ def _run_app_command(app_name: str, app_id: Optional[str]):
61
61
  if value == command and "_" in value:
62
62
  print(
63
63
  f"The command `{value}` was provided with an underscore and it isn't allowed."
64
- f"Instead, use `lightning {value.replace('_', ' ')}`."
64
+ f"Instead, use `lightning_app {value.replace('_', ' ')}`."
65
65
  )
66
66
  sys.exit(0)
67
67
  break
@@ -88,7 +88,7 @@ def _run_app_command(app_name: str, app_id: Optional[str]):
88
88
  def _handle_command_without_client(command: str, metadata: Dict, url: str) -> None:
89
89
  supported_params = list(metadata["parameters"])
90
90
  if _is_running_help(sys.argv):
91
- print(f"Usage: lightning {command} [ARGS]...")
91
+ print(f"Usage: lightning_app {command} [ARGS]...")
92
92
  print(" ")
93
93
  print("Options")
94
94
  for param in supported_params:
@@ -81,7 +81,7 @@ def ls(path: Optional[str] = None, print: bool = True, use_live: bool = True) ->
81
81
  # This happens if the user changes cluster and the project doesn't exit.
82
82
  if len(project) == 0:
83
83
  return _error_and_exit(
84
- f"There isn't any Lightning Project matching the name {splits[0]}." " HINT: Use `lightning cd`."
84
+ f"There isn't any Lightning Project matching the name {splits[0]}." " HINT: Use `lightning_app cd`."
85
85
  )
86
86
 
87
87
  project_id = project[0].project_id
@@ -3,7 +3,7 @@
3
3
  This ⚡ [Lightning component](https://lightning.ai/) ⚡ was generated automatically with:
4
4
 
5
5
  ```bash
6
- lightning init component placeholdername
6
+ lightning_app init component placeholdername
7
7
  ```
8
8
 
9
9
  ## To run placeholdername
@@ -11,7 +11,7 @@ lightning init component placeholdername
11
11
  First, install placeholdername (warning: this component has not been officially approved on the lightning gallery):
12
12
 
13
13
  ```bash
14
- lightning install component https://github.com/theUser/placeholdername
14
+ lightning_app install component https://github.com/theUser/placeholdername
15
15
  ```
16
16
 
17
17
  Once the app is installed, use it in an app:
@@ -5,7 +5,7 @@ from setuptools import find_packages, setup
5
5
  setup(
6
6
  name="placeholdername",
7
7
  version="0.0.0",
8
- description="⚡ Lightning component ⚡ generated with command: lightning init component",
8
+ description="⚡ Lightning component ⚡ generated with command: lightning_app init component",
9
9
  author="",
10
10
  author_email="",
11
11
  # REPLACE WITH YOUR OWN GITHUB PROJECT LINK
@@ -129,11 +129,11 @@ def connect_app(app_name_or_id: str):
129
129
  with open(connected_file, "w") as f:
130
130
  f.write(app_name_or_id + "\n")
131
131
 
132
- click.echo("The lightning CLI now responds to app commands. Use 'lightning --help' to see them.")
132
+ click.echo("The lightning App CLI now responds to app commands. Use 'lightning_app --help' to see them.")
133
133
  click.echo(" ")
134
134
 
135
135
  Popen(
136
- f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning --help",
136
+ f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning_app --help",
137
137
  shell=True,
138
138
  stdout=sys.stdout,
139
139
  stderr=sys.stderr,
@@ -147,11 +147,11 @@ def connect_app(app_name_or_id: str):
147
147
  shutil.copytree(matched_commands, commands)
148
148
  shutil.copy(matched_connected_file, connected_file)
149
149
 
150
- click.echo("The lightning CLI now responds to app commands. Use 'lightning --help' to see them.")
150
+ click.echo("The lightning App CLI now responds to app commands. Use 'lightning_app --help' to see them.")
151
151
  click.echo(" ")
152
152
 
153
153
  Popen(
154
- f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning --help",
154
+ f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning_app --help",
155
155
  shell=True,
156
156
  stdout=sys.stdout,
157
157
  stderr=sys.stderr,
@@ -205,11 +205,11 @@ def connect_app(app_name_or_id: str):
205
205
  f.write(retriever.app_name + "\n")
206
206
  f.write(retriever.app_id + "\n")
207
207
 
208
- click.echo("The lightning CLI now responds to app commands. Use 'lightning --help' to see them.")
208
+ click.echo("The lightning App CLI now responds to app commands. Use 'lightning_app --help' to see them.")
209
209
  click.echo(" ")
210
210
 
211
211
  Popen(
212
- f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning --help",
212
+ f"LIGHTNING_CONNECT_PPID={_PPID} {sys.executable} -m lightning_app --help",
213
213
  shell=True,
214
214
  stdout=sys.stdout,
215
215
  stderr=sys.stderr,
@@ -238,7 +238,7 @@ def disconnect_app(logout: bool = False):
238
238
  if not logout:
239
239
  click.echo(
240
240
  "You aren't connected to any Lightning App. "
241
- "Please use `lightning connect app_name_or_id` to connect to one."
241
+ "Please use `lightning_app connect app_name_or_id` to connect to one."
242
242
  )
243
243
 
244
244
 
@@ -104,7 +104,7 @@ def main() -> None:
104
104
  _run_app_command(app_name, app_id)
105
105
 
106
106
  click.echo()
107
- click.echo(message + " Return to the primary CLI with `lightning disconnect`.")
107
+ click.echo(message + " Return to the primary CLI with `lightning_app disconnect`.")
108
108
  else:
109
109
  _main()
110
110
 
@@ -5,7 +5,7 @@ This is a full react template ready to use in a component
5
5
  This UI was automatically generated with:
6
6
 
7
7
  ```commandline
8
- lightning init react-ui
8
+ lightning_app init react-ui
9
9
  ```
10
10
 
11
11
  ### Delete files
@@ -26,7 +26,7 @@ This template comes with `example_app.py` to show how to integrate the UI into a
26
26
  run it with:
27
27
 
28
28
  ```bash
29
- lightning run app react-ui/example_app.py
29
+ lightning_app run app react-ui/example_app.py
30
30
  ```
31
31
 
32
32
  ### Connect React to your component
@@ -716,11 +716,11 @@ class LightningFlow:
716
716
 
717
717
  .. code-block:: bash
718
718
 
719
- lightning run app app.py
719
+ lightning_app run app app.py
720
720
 
721
721
  .. code-block:: bash
722
722
 
723
- lightning my_command_name --args name=my_own_name
723
+ lightning_app my_command_name --args name=my_own_name
724
724
 
725
725
  """
726
726
  raise NotImplementedError
@@ -131,7 +131,7 @@ class CloudRuntime(Runtime):
131
131
  user = self.backend.client.auth_service_get_user()
132
132
  if not user.features.code_tab:
133
133
  rich.print(
134
- "[red]The `lightning open` command has not been enabled for your account. "
134
+ "[red]The `lightning_app open` command has not been enabled for your account. "
135
135
  "To request access, please contact support@lightning.ai[/red]"
136
136
  )
137
137
  sys.exit(1)
@@ -275,7 +275,7 @@ def run_app_in_cloud(
275
275
  token = res.json()["token"]
276
276
 
277
277
  # 3. Disconnect from the App if any.
278
- Popen("lightning logout", shell=True).wait()
278
+ Popen("lightning_app logout", shell=True).wait()
279
279
 
280
280
  # 4. Launch the application in the cloud from the Lightning CLI.
281
281
  with tempfile.TemporaryDirectory() as tmpdir:
@@ -239,7 +239,7 @@ def _patch_sys_argv():
239
239
  """This function modifies the ``sys.argv`` by extracting the arguments after ``--app_args`` and removed everything
240
240
  else before executing the user app script.
241
241
 
242
- The command: ``lightning run app app.py --without-server --app_args --use_gpu --env ...`` will be converted into
242
+ The command: ``lightning_app run app app.py --without-server --app_args --use_gpu --env ...`` will be converted into
243
243
  ``app.py --use_gpu``
244
244
 
245
245
  """
@@ -144,7 +144,7 @@ class Auth:
144
144
 
145
145
  raise ValueError(
146
146
  "We couldn't find any credentials linked to your account. "
147
- "Please try logging in using the CLI command `lightning login`"
147
+ "Please try logging in using the CLI command `lightning_app login`"
148
148
  )
149
149
 
150
150
 
@@ -1,7 +1,7 @@
1
+ from lightning.data.processing.functions import map, optimize, walk
1
2
  from lightning.data.streaming.combined import CombinedStreamingDataset
2
3
  from lightning.data.streaming.dataloader import StreamingDataLoader
3
4
  from lightning.data.streaming.dataset import StreamingDataset
4
- from lightning.data.streaming.functions import map, optimize, walk
5
5
 
6
6
  __all__ = [
7
7
  "LightningDataset",
@@ -57,3 +57,4 @@ _NUMPY_SCTYPES = [v for values in np.sctypes.values() for v in values]
57
57
  _NUMPY_DTYPES_MAPPING = {i: np.dtype(v) for i, v in enumerate(_NUMPY_SCTYPES)}
58
58
 
59
59
  _TIME_FORMAT = "%Y-%m-%d_%H-%M-%S.%fZ"
60
+ _IS_IN_STUDIO = bool(os.getenv("LIGHTNING_CLOUD_PROJECT_ID", None)) and bool(os.getenv("LIGHTNING_CLUSTER_ID", None))