lightning 2.2.3__tar.gz → 2.5.0.dev20241208__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.
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/.actions/assistant.py +38 -58
- {lightning-2.2.3/src/lightning.egg-info → lightning-2.5.0.dev20241208}/PKG-INFO +80 -109
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/README.md +83 -99
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/pyproject.toml +18 -117
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/base.txt +3 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/fabric/base.txt +3 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/fabric/examples.txt +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/fabric/strategies.txt +3 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/fabric/test.txt +2 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/pytorch/base.txt +4 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/pytorch/examples.txt +3 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/pytorch/extra.txt +4 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/pytorch/strategies.txt +1 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/pytorch/test.txt +3 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/setup.py +4 -5
- lightning-2.2.3/src/lightning/baef50ad7fe3d8b7db4846f46c5a01de6d9f12114b8a5f91 → lightning-2.5.0.dev20241208/src/lightning/0f6744a27141857e96e53b2891a5c697169ec2de5d04c60d +0 -8
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/__init__.py +0 -17
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/__setup__.py +9 -25
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/data/__init__.py +0 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/CHANGELOG.md +54 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/__init__.py +7 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/accelerators/cpu.py +5 -5
- lightning-2.5.0.dev20241208/src/lightning/fabric/accelerators/cuda.py +181 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/accelerators/mps.py +7 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/accelerators/registry.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/accelerators/xla.py +12 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/cli.py +50 -20
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/connector.py +19 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/fabric.py +60 -88
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/loggers/csv_logs.py +11 -11
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/loggers/logger.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/loggers/tensorboard.py +7 -11
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/collectives/collective.py +11 -9
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/collectives/single_device.py +7 -7
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/collectives/torch_collective.py +15 -15
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/lsf.py +2 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/io/checkpoint_io.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/io/torch_io.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/io/xla.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/amp.py +9 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/bitsandbytes.py +27 -21
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/deepspeed.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/double.py +5 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/fsdp.py +11 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/half.py +5 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/precision.py +7 -7
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/transformer_engine.py +6 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/utils.py +3 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/__init__.py +1 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/ddp.py +18 -8
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/deepspeed.py +31 -26
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/dp.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/fsdp.py +91 -160
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/launchers/multiprocessing.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/launchers/subprocess_script.py +6 -5
- lightning-2.5.0.dev20241208/src/lightning/fabric/strategies/model_parallel.py +595 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/parallel.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/registry.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/single_xla.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/strategy.py +22 -24
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/xla.py +6 -6
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/xla_fsdp.py +24 -33
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/apply_func.py +13 -6
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/cloud_io.py +14 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/consolidate_checkpoint.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/data.py +15 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/device_dtype_mixin.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/device_parser.py +11 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/distributed.py +61 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/imports.py +8 -6
- lightning-2.5.0.dev20241208/src/lightning/fabric/utilities/init.py +114 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/load.py +24 -68
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/logger.py +23 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/optimizer.py +11 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/rank_zero.py +4 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/registry.py +10 -16
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/seed.py +50 -23
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/spike.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/testing/_runif.py +9 -12
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/throughput.py +15 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/types.py +13 -59
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/warnings.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/wrappers.py +70 -45
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/CHANGELOG.md +62 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/__init__.py +1 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/_graveyard/__init__.py +0 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/accelerators/accelerator.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/accelerators/cpu.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/accelerators/cuda.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/accelerators/mps.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/accelerators/xla.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/callback.py +6 -6
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/device_stats_monitor.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/early_stopping.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/finetuning.py +11 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/gradient_accumulation_scheduler.py +3 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/lr_monitor.py +28 -26
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/model_checkpoint.py +22 -21
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/model_summary.py +14 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/prediction_writer.py +2 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/progress/progress_bar.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/progress/rich_progress.py +26 -53
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +11 -6
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/pruning.py +16 -15
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/rich_model_summary.py +6 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/spike.py +2 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/stochastic_weight_avg.py +11 -11
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/throughput_monitor.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/timer.py +22 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/cli.py +115 -44
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/datamodule.py +14 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/hooks.py +8 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/mixins/hparams_mixin.py +22 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/module.py +54 -72
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/optimizer.py +16 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/saving.py +17 -10
- lightning-2.5.0.dev20241208/src/lightning/pytorch/demos/__init__.py +15 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/demos/boring_classes.py +5 -4
- lightning-2.5.0.dev20241208/src/lightning/pytorch/demos/lstm.py +99 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/demos/mnist_datamodule.py +4 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/demos/transformer.py +12 -13
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/comet.py +5 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/csv_logs.py +7 -9
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/logger.py +7 -8
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/mlflow.py +17 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/neptune.py +30 -46
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/tensorboard.py +4 -8
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/utilities.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/wandb.py +15 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/evaluation_loop.py +48 -11
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/fetchers.py +3 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/fit_loop.py +103 -12
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/loop.py +16 -6
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/optimization/automatic.py +13 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/optimization/closure.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/optimization/manual.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/prediction_loop.py +8 -7
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/progress.py +23 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/training_epoch_loop.py +82 -7
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/utilities.py +9 -12
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/overrides/distributed.py +8 -7
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/io/wrapper.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/amp.py +9 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/bitsandbytes.py +1 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/deepspeed.py +6 -6
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/double.py +5 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/fsdp.py +11 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/half.py +5 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/precision.py +5 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/profilers/advanced.py +29 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/profilers/profiler.py +3 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/profilers/pytorch.py +33 -15
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/profilers/simple.py +8 -8
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/profilers/xla.py +2 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/serve/servable_module.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/serve/servable_module_validator.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/__init__.py +2 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/ddp.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/deepspeed.py +19 -21
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/fsdp.py +59 -67
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/launchers/multiprocessing.py +19 -17
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/launchers/subprocess_script.py +3 -3
- lightning-2.5.0.dev20241208/src/lightning/pytorch/strategies/model_parallel.py +364 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/parallel.py +6 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/single_xla.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/strategy.py +17 -18
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/xla.py +6 -6
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/call.py +37 -23
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/configuration_validator.py +0 -20
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +28 -83
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/callback_connector.py +10 -9
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/checkpoint_connector.py +5 -7
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/data_connector.py +15 -31
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/logger_connector/fx_validator.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/logger_connector/logger_connector.py +2 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/logger_connector/result.py +15 -22
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/signal_connector.py +16 -35
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/setup.py +1 -17
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/trainer.py +31 -45
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/tuner/batch_size_scaling.py +7 -7
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/tuner/lr_finder.py +24 -23
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/__init__.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/_pytree.py +5 -5
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/argparse.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/combined_loader.py +19 -19
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/compile.py +2 -23
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/consolidate_checkpoint.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/data.py +15 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/grads.py +2 -2
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/imports.py +0 -11
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/migration/migration.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/migration/utils.py +4 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/model_helpers.py +11 -8
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/model_summary/model_summary.py +66 -42
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py +10 -10
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/parameter_tying.py +3 -3
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/parsing.py +24 -14
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/seed.py +1 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/testing/_runif.py +3 -4
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/types.py +12 -17
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/upgrade_checkpoint.py +1 -2
- lightning-2.5.0.dev20241208/src/lightning/version.info +1 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208/src/lightning.egg-info}/PKG-INFO +80 -109
- lightning-2.5.0.dev20241208/src/lightning.egg-info/SOURCES.txt +294 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning.egg-info/entry_points.txt +0 -1
- lightning-2.5.0.dev20241208/src/lightning.egg-info/requires.txt +297 -0
- lightning-2.5.0.dev20241208/src/version.info +1 -0
- lightning-2.2.3/requirements/app/app.txt +0 -29
- lightning-2.2.3/requirements/app/cloud.txt +0 -4
- lightning-2.2.3/requirements/app/components.txt +0 -5
- lightning-2.2.3/requirements/app/docs.txt +0 -1
- lightning-2.2.3/requirements/app/test.txt +0 -18
- lightning-2.2.3/requirements/app/ui.txt +0 -0
- lightning-2.2.3/src/lightning/__main__.py +0 -4
- lightning-2.2.3/src/lightning/app/CHANGELOG.md +0 -608
- lightning-2.2.3/src/lightning/app/__init__.py +0 -51
- lightning-2.2.3/src/lightning/app/api/__init__.py +0 -8
- lightning-2.2.3/src/lightning/app/api/http_methods.py +0 -258
- lightning-2.2.3/src/lightning/app/api/request_types.py +0 -56
- lightning-2.2.3/src/lightning/app/cli/app-template/.gitignore +0 -157
- lightning-2.2.3/src/lightning/app/cli/app-template/LICENSE +0 -201
- lightning-2.2.3/src/lightning/app/cli/app-template/README.md +0 -37
- lightning-2.2.3/src/lightning/app/cli/app-template/app.py +0 -16
- lightning-2.2.3/src/lightning/app/cli/app-template/placeholdername/__init__.py +0 -4
- lightning-2.2.3/src/lightning/app/cli/app-template/placeholdername/components/component_a/__init__.py +0 -3
- lightning-2.2.3/src/lightning/app/cli/app-template/placeholdername/components/component_a/component_a.py +0 -6
- lightning-2.2.3/src/lightning/app/cli/app-template/placeholdername/components/component_b/__init__.py +0 -3
- lightning-2.2.3/src/lightning/app/cli/app-template/placeholdername/components/component_b/component_a.py +0 -6
- lightning-2.2.3/src/lightning/app/cli/app-template/requirements.txt +0 -0
- lightning-2.2.3/src/lightning/app/cli/app-template/setup.py +0 -15
- lightning-2.2.3/src/lightning/app/cli/app-template/tests/README.md +0 -17
- lightning-2.2.3/src/lightning/app/cli/app-template/tests/requirements.txt +0 -8
- lightning-2.2.3/src/lightning/app/cli/app-template/tests/test_placeholdername_app.py +0 -44
- lightning-2.2.3/src/lightning/app/cli/cmd_apps.py +0 -146
- lightning-2.2.3/src/lightning/app/cli/cmd_init.py +0 -167
- lightning-2.2.3/src/lightning/app/cli/cmd_install.py +0 -657
- lightning-2.2.3/src/lightning/app/cli/cmd_pl_init.py +0 -187
- lightning-2.2.3/src/lightning/app/cli/cmd_react_ui_init.py +0 -131
- lightning-2.2.3/src/lightning/app/cli/commands/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/cli/commands/app_commands.py +0 -135
- lightning-2.2.3/src/lightning/app/cli/commands/cd.py +0 -117
- lightning-2.2.3/src/lightning/app/cli/commands/cp.py +0 -350
- lightning-2.2.3/src/lightning/app/cli/commands/logs.py +0 -122
- lightning-2.2.3/src/lightning/app/cli/commands/ls.py +0 -268
- lightning-2.2.3/src/lightning/app/cli/commands/pwd.py +0 -53
- lightning-2.2.3/src/lightning/app/cli/commands/rm.py +0 -101
- lightning-2.2.3/src/lightning/app/cli/component-template/.github/workflows/ci-testing.yml +0 -79
- lightning-2.2.3/src/lightning/app/cli/component-template/.gitignore +0 -157
- lightning-2.2.3/src/lightning/app/cli/component-template/LICENSE +0 -201
- lightning-2.2.3/src/lightning/app/cli/component-template/README.md +0 -35
- lightning-2.2.3/src/lightning/app/cli/component-template/app.py +0 -15
- lightning-2.2.3/src/lightning/app/cli/component-template/placeholdername/__init__.py +0 -3
- lightning-2.2.3/src/lightning/app/cli/component-template/placeholdername/component.py +0 -12
- lightning-2.2.3/src/lightning/app/cli/component-template/requirements.txt +0 -0
- lightning-2.2.3/src/lightning/app/cli/component-template/setup.py +0 -15
- lightning-2.2.3/src/lightning/app/cli/component-template/tests/README.md +0 -17
- lightning-2.2.3/src/lightning/app/cli/component-template/tests/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/cli/component-template/tests/requirements.txt +0 -8
- lightning-2.2.3/src/lightning/app/cli/component-template/tests/test_placeholdername_component.py +0 -14
- lightning-2.2.3/src/lightning/app/cli/connect/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/cli/connect/app.py +0 -387
- lightning-2.2.3/src/lightning/app/cli/connect/data.py +0 -109
- lightning-2.2.3/src/lightning/app/cli/core.py +0 -27
- lightning-2.2.3/src/lightning/app/cli/lightning_cli.py +0 -403
- lightning-2.2.3/src/lightning/app/cli/lightning_cli_delete.py +0 -124
- lightning-2.2.3/src/lightning/app/cli/lightning_cli_launch.py +0 -130
- lightning-2.2.3/src/lightning/app/cli/lightning_cli_list.py +0 -32
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/.gitignore +0 -1
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/.lightningignore +0 -2
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/app.py +0 -105
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/callbacks.py +0 -319
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/components/__init__.py +0 -2
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/components/logger/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/components/logger/tensorboard.py +0 -49
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/components/logger/weights_and_biases.py +0 -33
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/components/script_runner/__init__.py +0 -1
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/components/script_runner/script_runner.py +0 -76
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/core/state.py +0 -45
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/setup.py +0 -34
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/tests/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/tests/core/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/tests/core/test_callbacks.py +0 -68
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/tests/test_app.py +0 -14
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/.gitignore +0 -25
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/.prettierignore +0 -3
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/.prettierrc +0 -24
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/craco.config.js +0 -29
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/package.json +0 -95
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/public/favicon.svg +0 -9
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/public/index.html +0 -65
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/public/manifest.json +0 -15
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/public/robots.txt +0 -3
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/App.tsx +0 -126
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/EnvironmentConfigurator.tsx +0 -67
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/ErrorPanel.tsx +0 -24
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/ExecutionSummary.tsx +0 -79
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/HyperparameterSummary.tsx +0 -95
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/Launcher.tsx +0 -172
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/ProgressBar.tsx +0 -35
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/ProgressBarGroup.tsx +0 -49
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/components/Timer.tsx +0 -29
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/hooks/useLightningState.ts +0 -31
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/index.css +0 -19
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/index.tsx +0 -26
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/lightning-colors.ts +0 -2
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/react-app-env.d.ts +0 -1
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/reportWebVitals.ts +0 -15
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/src/types/lightning.ts +0 -57
- lightning-2.2.3/src/lightning/app/cli/pl-app-template/ui/tsconfig.json +0 -22
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/README.md +0 -103
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/example_app.py +0 -33
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/index.html +0 -14
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/package.json +0 -31
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/App.css +0 -10
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/App.tsx +0 -69
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/favicon.svg +0 -15
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/hooks/useLightningState.ts +0 -31
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/index.css +0 -11
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/main.tsx +0 -11
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/types/lightning.ts +0 -57
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/src/vite-env.d.ts +0 -1
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/tsconfig.json +0 -21
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/tsconfig.node.json +0 -8
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/vite.config.ts +0 -9
- lightning-2.2.3/src/lightning/app/cli/react-ui-template/ui/yarn.lock +0 -1278
- lightning-2.2.3/src/lightning/app/components/__init__.py +0 -41
- lightning-2.2.3/src/lightning/app/components/database/__init__.py +0 -4
- lightning-2.2.3/src/lightning/app/components/database/client.py +0 -93
- lightning-2.2.3/src/lightning/app/components/database/server.py +0 -243
- lightning-2.2.3/src/lightning/app/components/database/utilities.py +0 -263
- lightning-2.2.3/src/lightning/app/components/multi_node/__init__.py +0 -6
- lightning-2.2.3/src/lightning/app/components/multi_node/base.py +0 -107
- lightning-2.2.3/src/lightning/app/components/multi_node/fabric.py +0 -131
- lightning-2.2.3/src/lightning/app/components/multi_node/pytorch_spawn.py +0 -119
- lightning-2.2.3/src/lightning/app/components/multi_node/trainer.py +0 -129
- lightning-2.2.3/src/lightning/app/components/python/__init__.py +0 -4
- lightning-2.2.3/src/lightning/app/components/python/popen.py +0 -113
- lightning-2.2.3/src/lightning/app/components/python/tracer.py +0 -200
- lightning-2.2.3/src/lightning/app/components/serve/__init__.py +0 -17
- lightning-2.2.3/src/lightning/app/components/serve/auto_scaler.py +0 -753
- lightning-2.2.3/src/lightning/app/components/serve/catimage.png +0 -0
- lightning-2.2.3/src/lightning/app/components/serve/cold_start_proxy.py +0 -72
- lightning-2.2.3/src/lightning/app/components/serve/gradio_server.py +0 -200
- lightning-2.2.3/src/lightning/app/components/serve/python_server.py +0 -328
- lightning-2.2.3/src/lightning/app/components/serve/serve.py +0 -170
- lightning-2.2.3/src/lightning/app/components/serve/streamlit.py +0 -174
- lightning-2.2.3/src/lightning/app/components/serve/types/__init__.py +0 -4
- lightning-2.2.3/src/lightning/app/components/serve/types/image.py +0 -45
- lightning-2.2.3/src/lightning/app/components/serve/types/type.py +0 -32
- lightning-2.2.3/src/lightning/app/components/training.py +0 -203
- lightning-2.2.3/src/lightning/app/core/__init__.py +0 -5
- lightning-2.2.3/src/lightning/app/core/api.py +0 -498
- lightning-2.2.3/src/lightning/app/core/app.py +0 -737
- lightning-2.2.3/src/lightning/app/core/constants.py +0 -125
- lightning-2.2.3/src/lightning/app/core/flow.py +0 -861
- lightning-2.2.3/src/lightning/app/core/queues.py +0 -574
- lightning-2.2.3/src/lightning/app/core/work.py +0 -769
- lightning-2.2.3/src/lightning/app/frontend/__init__.py +0 -7
- lightning-2.2.3/src/lightning/app/frontend/frontend.py +0 -68
- lightning-2.2.3/src/lightning/app/frontend/just_py/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/frontend/just_py/just_py.py +0 -114
- lightning-2.2.3/src/lightning/app/frontend/just_py/just_py_base.py +0 -66
- lightning-2.2.3/src/lightning/app/frontend/panel/__init__.py +0 -6
- lightning-2.2.3/src/lightning/app/frontend/panel/app_state_comm.py +0 -101
- lightning-2.2.3/src/lightning/app/frontend/panel/app_state_watcher.py +0 -123
- lightning-2.2.3/src/lightning/app/frontend/panel/panel_frontend.py +0 -193
- lightning-2.2.3/src/lightning/app/frontend/panel/panel_serve_render_fn.py +0 -71
- lightning-2.2.3/src/lightning/app/frontend/stream_lit.py +0 -113
- lightning-2.2.3/src/lightning/app/frontend/streamlit_base.py +0 -50
- lightning-2.2.3/src/lightning/app/frontend/utils.py +0 -72
- lightning-2.2.3/src/lightning/app/frontend/web.py +0 -140
- lightning-2.2.3/src/lightning/app/launcher/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/launcher/launcher.py +0 -440
- lightning-2.2.3/src/lightning/app/launcher/lightning_backend.py +0 -525
- lightning-2.2.3/src/lightning/app/launcher/lightning_hybrid_backend.py +0 -155
- lightning-2.2.3/src/lightning/app/pdb/__init__.py +0 -6
- lightning-2.2.3/src/lightning/app/pdb/pdb.py +0 -50
- lightning-2.2.3/src/lightning/app/plugin/__init__.py +0 -3
- lightning-2.2.3/src/lightning/app/plugin/plugin.py +0 -237
- lightning-2.2.3/src/lightning/app/runners/__init__.py +0 -14
- lightning-2.2.3/src/lightning/app/runners/backends/__init__.py +0 -24
- lightning-2.2.3/src/lightning/app/runners/backends/backend.py +0 -143
- lightning-2.2.3/src/lightning/app/runners/backends/cloud.py +0 -49
- lightning-2.2.3/src/lightning/app/runners/backends/docker.py +0 -40
- lightning-2.2.3/src/lightning/app/runners/backends/mp_process.py +0 -137
- lightning-2.2.3/src/lightning/app/runners/cloud.py +0 -1109
- lightning-2.2.3/src/lightning/app/runners/multiprocess.py +0 -160
- lightning-2.2.3/src/lightning/app/runners/runtime.py +0 -182
- lightning-2.2.3/src/lightning/app/runners/runtime_type.py +0 -33
- lightning-2.2.3/src/lightning/app/source_code/__init__.py +0 -7
- lightning-2.2.3/src/lightning/app/source_code/copytree.py +0 -183
- lightning-2.2.3/src/lightning/app/source_code/hashing.py +0 -51
- lightning-2.2.3/src/lightning/app/source_code/local.py +0 -149
- lightning-2.2.3/src/lightning/app/source_code/tar.py +0 -201
- lightning-2.2.3/src/lightning/app/source_code/uploader.py +0 -111
- lightning-2.2.3/src/lightning/app/storage/__init__.py +0 -6
- lightning-2.2.3/src/lightning/app/storage/copier.py +0 -155
- lightning-2.2.3/src/lightning/app/storage/drive.py +0 -341
- lightning-2.2.3/src/lightning/app/storage/filesystem.py +0 -166
- lightning-2.2.3/src/lightning/app/storage/mount.py +0 -75
- lightning-2.2.3/src/lightning/app/storage/orchestrator.py +0 -208
- lightning-2.2.3/src/lightning/app/storage/path.py +0 -453
- lightning-2.2.3/src/lightning/app/storage/payload.py +0 -274
- lightning-2.2.3/src/lightning/app/storage/requests.py +0 -57
- lightning-2.2.3/src/lightning/app/structures/__init__.py +0 -4
- lightning-2.2.3/src/lightning/app/structures/dict.py +0 -160
- lightning-2.2.3/src/lightning/app/structures/list.py +0 -177
- lightning-2.2.3/src/lightning/app/testing/__init__.py +0 -20
- lightning-2.2.3/src/lightning/app/testing/config.py +0 -28
- lightning-2.2.3/src/lightning/app/testing/helpers.py +0 -179
- lightning-2.2.3/src/lightning/app/testing/testing.py +0 -535
- lightning-2.2.3/src/lightning/app/ui/asset-manifest.json +0 -19
- lightning-2.2.3/src/lightning/app/ui/index.html +0 -1
- lightning-2.2.3/src/lightning/app/ui/static/css/main.fb7060be.css +0 -2
- lightning-2.2.3/src/lightning/app/ui/static/css/main.fb7060be.css.map +0 -1
- lightning-2.2.3/src/lightning/app/ui/static/favicon.ico +0 -0
- lightning-2.2.3/src/lightning/app/ui/static/js/787.418637a8.chunk.js +0 -2
- lightning-2.2.3/src/lightning/app/ui/static/js/787.418637a8.chunk.js.map +0 -1
- lightning-2.2.3/src/lightning/app/ui/static/js/main.c1f02aeb.js +0 -3
- lightning-2.2.3/src/lightning/app/ui/static/js/main.c1f02aeb.js.LICENSE.txt +0 -87
- lightning-2.2.3/src/lightning/app/ui/static/js/main.c1f02aeb.js.map +0 -1
- lightning-2.2.3/src/lightning/app/ui/static/lightningState.js +0 -24
- lightning-2.2.3/src/lightning/app/ui/static/manifest.json +0 -15
- lightning-2.2.3/src/lightning/app/ui/static/media/error.11892047d0183a4723b91dc0fb98cb95.svg +0 -5
- lightning-2.2.3/src/lightning/app/ui/static/media/lightning-logo-with-text.b964c8fbf221c97eb8ce52bb6e3a30d6.svg +0 -16
- lightning-2.2.3/src/lightning/app/ui/static/media/success.5edae4c5b171e2c1c5a3c54273c47ba8.svg +0 -4
- lightning-2.2.3/src/lightning/app/ui/static/media/warning.5c542673e84e77ceb6a230bfea7f7263.svg +0 -4
- lightning-2.2.3/src/lightning/app/ui/static/robots.txt +0 -3
- lightning-2.2.3/src/lightning/app/ui/version.info +0 -1
- lightning-2.2.3/src/lightning/app/utilities/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/utilities/app_commands.py +0 -127
- lightning-2.2.3/src/lightning/app/utilities/app_helpers.py +0 -582
- lightning-2.2.3/src/lightning/app/utilities/app_logs.py +0 -137
- lightning-2.2.3/src/lightning/app/utilities/app_status.py +0 -44
- lightning-2.2.3/src/lightning/app/utilities/auth.py +0 -61
- lightning-2.2.3/src/lightning/app/utilities/cli_helpers.py +0 -358
- lightning-2.2.3/src/lightning/app/utilities/cloud.py +0 -63
- lightning-2.2.3/src/lightning/app/utilities/clusters.py +0 -52
- lightning-2.2.3/src/lightning/app/utilities/commands/__init__.py +0 -3
- lightning-2.2.3/src/lightning/app/utilities/commands/base.py +0 -308
- lightning-2.2.3/src/lightning/app/utilities/component.py +0 -152
- lightning-2.2.3/src/lightning/app/utilities/data_structures.py +0 -50
- lightning-2.2.3/src/lightning/app/utilities/dependency_caching.py +0 -27
- lightning-2.2.3/src/lightning/app/utilities/enum.py +0 -82
- lightning-2.2.3/src/lightning/app/utilities/exceptions.py +0 -101
- lightning-2.2.3/src/lightning/app/utilities/frontend.py +0 -89
- lightning-2.2.3/src/lightning/app/utilities/git.py +0 -86
- lightning-2.2.3/src/lightning/app/utilities/imports.py +0 -148
- lightning-2.2.3/src/lightning/app/utilities/introspection.py +0 -400
- lightning-2.2.3/src/lightning/app/utilities/layout.py +0 -215
- lightning-2.2.3/src/lightning/app/utilities/load_app.py +0 -304
- lightning-2.2.3/src/lightning/app/utilities/log.py +0 -23
- lightning-2.2.3/src/lightning/app/utilities/log_helpers.py +0 -49
- lightning-2.2.3/src/lightning/app/utilities/login.py +0 -213
- lightning-2.2.3/src/lightning/app/utilities/logs_socket_api.py +0 -94
- lightning-2.2.3/src/lightning/app/utilities/name_generator.py +0 -1359
- lightning-2.2.3/src/lightning/app/utilities/network.py +0 -211
- lightning-2.2.3/src/lightning/app/utilities/openapi.py +0 -75
- lightning-2.2.3/src/lightning/app/utilities/packaging/__init__.py +0 -0
- lightning-2.2.3/src/lightning/app/utilities/packaging/app_config.py +0 -76
- lightning-2.2.3/src/lightning/app/utilities/packaging/build_config.py +0 -208
- lightning-2.2.3/src/lightning/app/utilities/packaging/cloud_compute.py +0 -188
- lightning-2.2.3/src/lightning/app/utilities/packaging/docker.py +0 -127
- lightning-2.2.3/src/lightning/app/utilities/packaging/lightning_utils.py +0 -219
- lightning-2.2.3/src/lightning/app/utilities/packaging/tarfile.py +0 -52
- lightning-2.2.3/src/lightning/app/utilities/port.py +0 -168
- lightning-2.2.3/src/lightning/app/utilities/proxies.py +0 -745
- lightning-2.2.3/src/lightning/app/utilities/redis.py +0 -34
- lightning-2.2.3/src/lightning/app/utilities/safe_pickle.py +0 -107
- lightning-2.2.3/src/lightning/app/utilities/scheduler.py +0 -60
- lightning-2.2.3/src/lightning/app/utilities/secrets.py +0 -41
- lightning-2.2.3/src/lightning/app/utilities/state.py +0 -323
- lightning-2.2.3/src/lightning/app/utilities/tracer.py +0 -193
- lightning-2.2.3/src/lightning/app/utilities/tree.py +0 -88
- lightning-2.2.3/src/lightning/app/utilities/types.py +0 -27
- lightning-2.2.3/src/lightning/app/utilities/warnings.py +0 -17
- lightning-2.2.3/src/lightning/fabric/accelerators/cuda.py +0 -381
- lightning-2.2.3/src/lightning/fabric/utilities/init.py +0 -76
- lightning-2.2.3/src/lightning/pytorch/_graveyard/ipu.py +0 -52
- lightning-2.2.3/src/lightning/pytorch/demos/__init__.py +0 -1
- lightning-2.2.3/src/lightning/pytorch/overrides/__init__.py +0 -0
- lightning-2.2.3/src/lightning/pytorch/trainer/connectors/__init__.py +0 -0
- lightning-2.2.3/src/lightning/store/__init__.py +0 -3
- lightning-2.2.3/src/lightning/store/store.py +0 -92
- lightning-2.2.3/src/lightning/store/utils.py +0 -67
- lightning-2.2.3/src/lightning/version.info +0 -1
- lightning-2.2.3/src/lightning.egg-info/SOURCES.txt +0 -567
- lightning-2.2.3/src/lightning.egg-info/requires.txt +0 -595
- lightning-2.2.3/src/version.info +0 -1
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/CITATION.cff +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/LICENSE +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/fabric/docs.txt +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/requirements/pytorch/docs.txt +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/setup.cfg +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/__about__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/__version__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/_graveyard/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/_graveyard/tpu.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/accelerators/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/accelerators/accelerator.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/loggers/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/collectives/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/cluster_environment.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/kubeflow.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/lightning.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/mpi.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/slurm.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/torchelastic.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/environments/xla.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/io/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/plugins/precision/xla.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/launchers/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/launchers/launcher.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/launchers/xla.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/strategies/single_device.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/enums.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/exceptions.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/fabric/utilities/testing/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/py.typed +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/_graveyard/_torchmetrics.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/_graveyard/hpu.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/_graveyard/precision.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/_graveyard/tpu.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/accelerators/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/batch_size_finder.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/checkpoint.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/lambda_function.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/lr_finder.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/on_exception_checkpoint.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/callbacks/progress/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/core/mixins/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loggers/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/loops/optimization/__init__.py +0 -0
- {lightning-2.2.3/src/lightning/app/cli → lightning-2.5.0.dev20241208/src/lightning/pytorch/overrides}/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/environments/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/io/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/io/async_plugin.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/io/checkpoint_plugin.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/io/torch_plugin.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/io/xla_plugin.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/layer_sync.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/transformer_engine.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/plugins/precision/xla.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/profilers/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/profilers/base.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/serve/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/launchers/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/launchers/launcher.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/launchers/xla.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/strategies/single_device.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/__init__.py +0 -0
- {lightning-2.2.3/src/lightning/app/cli/app-template/tests → lightning-2.5.0.dev20241208/src/lightning/pytorch/trainer/connectors}/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/connectors/logger_connector/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/trainer/states.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/tuner/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/tuner/tuning.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/deepspeed.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/enums.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/exceptions.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/memory.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/migration/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/model_summary/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/rank_zero.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/signature_utils.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/testing/__init__.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning/pytorch/utilities/warnings.py +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning.egg-info/dependency_links.txt +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning.egg-info/not-zip-safe +0 -0
- {lightning-2.2.3 → lightning-2.5.0.dev20241208}/src/lightning.egg-info/top_level.txt +0 -0
|
@@ -14,19 +14,18 @@
|
|
|
14
14
|
import glob
|
|
15
15
|
import logging
|
|
16
16
|
import os
|
|
17
|
-
import pathlib
|
|
18
17
|
import re
|
|
19
18
|
import shutil
|
|
20
|
-
import tarfile
|
|
21
19
|
import tempfile
|
|
22
20
|
import urllib.request
|
|
23
|
-
from
|
|
21
|
+
from collections.abc import Iterable, Iterator, Sequence
|
|
24
22
|
from itertools import chain
|
|
25
23
|
from os.path import dirname, isfile
|
|
26
24
|
from pathlib import Path
|
|
27
|
-
from typing import Any,
|
|
25
|
+
from typing import Any, Optional
|
|
28
26
|
|
|
29
|
-
from
|
|
27
|
+
from packaging.requirements import Requirement
|
|
28
|
+
from packaging.version import Version
|
|
30
29
|
|
|
31
30
|
REQUIREMENT_FILES = {
|
|
32
31
|
"pytorch": (
|
|
@@ -35,11 +34,6 @@ REQUIREMENT_FILES = {
|
|
|
35
34
|
"requirements/pytorch/strategies.txt",
|
|
36
35
|
"requirements/pytorch/examples.txt",
|
|
37
36
|
),
|
|
38
|
-
"app": (
|
|
39
|
-
"requirements/app/app.txt",
|
|
40
|
-
"requirements/app/cloud.txt",
|
|
41
|
-
"requirements/app/ui.txt",
|
|
42
|
-
),
|
|
43
37
|
"fabric": (
|
|
44
38
|
"requirements/fabric/base.txt",
|
|
45
39
|
"requirements/fabric/strategies.txt",
|
|
@@ -87,14 +81,15 @@ class _RequirementWithComment(Requirement):
|
|
|
87
81
|
out = str(self)
|
|
88
82
|
if self.strict:
|
|
89
83
|
return f"{out} {self.strict_string}"
|
|
84
|
+
specs = [(spec.operator, spec.version) for spec in self.specifier]
|
|
90
85
|
if unfreeze == "major":
|
|
91
|
-
for operator, version in
|
|
86
|
+
for operator, version in specs:
|
|
92
87
|
if operator in ("<", "<="):
|
|
93
|
-
major =
|
|
88
|
+
major = Version(version).major
|
|
94
89
|
# replace upper bound with major version increased by one
|
|
95
90
|
return out.replace(f"{operator}{version}", f"<{major + 1}.0")
|
|
96
91
|
elif unfreeze == "all":
|
|
97
|
-
for operator, version in
|
|
92
|
+
for operator, version in specs:
|
|
98
93
|
if operator in ("<", "<="):
|
|
99
94
|
# drop upper bound
|
|
100
95
|
return out.replace(f"{operator}{version},", "")
|
|
@@ -103,33 +98,25 @@ class _RequirementWithComment(Requirement):
|
|
|
103
98
|
return out
|
|
104
99
|
|
|
105
100
|
|
|
106
|
-
def _parse_requirements(
|
|
101
|
+
def _parse_requirements(lines: Iterable[str]) -> Iterator[_RequirementWithComment]:
|
|
107
102
|
"""Adapted from `pkg_resources.parse_requirements` to include comments.
|
|
108
103
|
|
|
109
104
|
>>> txt = ['# ignored', '', 'this # is an', '--piparg', 'example', 'foo # strict', 'thing', '-r different/file.txt']
|
|
110
105
|
>>> [r.adjust('none') for r in _parse_requirements(txt)]
|
|
111
106
|
['this', 'example', 'foo # strict', 'thing']
|
|
112
|
-
>>> txt = '\\n'.join(txt)
|
|
113
|
-
>>> [r.adjust('none') for r in _parse_requirements(txt)]
|
|
114
|
-
['this', 'example', 'foo # strict', 'thing']
|
|
115
107
|
|
|
116
108
|
"""
|
|
117
|
-
lines = yield_lines(strs)
|
|
118
109
|
pip_argument = None
|
|
119
110
|
for line in lines:
|
|
111
|
+
line = line.strip()
|
|
112
|
+
if not line or line.startswith("#"):
|
|
113
|
+
continue
|
|
120
114
|
# Drop comments -- a hash without a space may be in a URL.
|
|
121
115
|
if " #" in line:
|
|
122
116
|
comment_pos = line.find(" #")
|
|
123
117
|
line, comment = line[:comment_pos], line[comment_pos:]
|
|
124
118
|
else:
|
|
125
119
|
comment = ""
|
|
126
|
-
# If there is a line continuation, drop it, and append the next line.
|
|
127
|
-
if line.endswith("\\"):
|
|
128
|
-
line = line[:-2].strip()
|
|
129
|
-
try:
|
|
130
|
-
line += next(lines)
|
|
131
|
-
except StopIteration:
|
|
132
|
-
return
|
|
133
120
|
# If there's a pip argument, save it
|
|
134
121
|
if line.startswith("--"):
|
|
135
122
|
pip_argument = line
|
|
@@ -141,7 +128,7 @@ def _parse_requirements(strs: Union[str, Iterable[str]]) -> Iterator[_Requiremen
|
|
|
141
128
|
pip_argument = None
|
|
142
129
|
|
|
143
130
|
|
|
144
|
-
def load_requirements(path_dir: str, file_name: str = "base.txt", unfreeze: str = "all") ->
|
|
131
|
+
def load_requirements(path_dir: str, file_name: str = "base.txt", unfreeze: str = "all") -> list[str]:
|
|
145
132
|
"""Loading requirements from a file.
|
|
146
133
|
|
|
147
134
|
>>> path_req = os.path.join(_PROJECT_ROOT, "requirements")
|
|
@@ -155,7 +142,7 @@ def load_requirements(path_dir: str, file_name: str = "base.txt", unfreeze: str
|
|
|
155
142
|
logging.warning(f"Folder {path_dir} does not have any base requirements.")
|
|
156
143
|
return []
|
|
157
144
|
assert path.exists(), (path_dir, file_name, path)
|
|
158
|
-
text = path.read_text()
|
|
145
|
+
text = path.read_text().splitlines()
|
|
159
146
|
return [req.adjust(unfreeze) for req in _parse_requirements(text)]
|
|
160
147
|
|
|
161
148
|
|
|
@@ -216,30 +203,6 @@ def distribute_version(src_folder: str, ver_file: str = "version.info") -> None:
|
|
|
216
203
|
shutil.copy2(ver_template, fpath)
|
|
217
204
|
|
|
218
205
|
|
|
219
|
-
def _download_frontend(pkg_path: str, version: str = "v0.0.0"):
|
|
220
|
-
"""Downloads an archive file for a specific release of the Lightning frontend and extracts it to the correct
|
|
221
|
-
directory."""
|
|
222
|
-
|
|
223
|
-
try:
|
|
224
|
-
frontend_dir = pathlib.Path(pkg_path, "ui")
|
|
225
|
-
download_dir = tempfile.mkdtemp()
|
|
226
|
-
|
|
227
|
-
shutil.rmtree(frontend_dir, ignore_errors=True)
|
|
228
|
-
# TODO: remove this once lightning-ui package is ready as a dependency
|
|
229
|
-
frontend_release_url = f"https://lightning-packages.s3.amazonaws.com/ui/{version}.tar.gz"
|
|
230
|
-
response = urllib.request.urlopen(frontend_release_url)
|
|
231
|
-
|
|
232
|
-
file = tarfile.open(fileobj=response, mode="r|gz")
|
|
233
|
-
file.extractall(path=download_dir) # noqa: S202
|
|
234
|
-
|
|
235
|
-
shutil.move(download_dir, frontend_dir)
|
|
236
|
-
print("The Lightning UI has successfully been downloaded!")
|
|
237
|
-
|
|
238
|
-
# If installing from source without internet connection, we don't want to break the installation
|
|
239
|
-
except Exception:
|
|
240
|
-
print("The Lightning UI downloading has failed!")
|
|
241
|
-
|
|
242
|
-
|
|
243
206
|
def _load_aggregate_requirements(req_dir: str = "requirements", freeze_requirements: bool = False) -> None:
|
|
244
207
|
"""Load all base requirements from all particular packages and prune duplicates.
|
|
245
208
|
|
|
@@ -260,7 +223,7 @@ def _load_aggregate_requirements(req_dir: str = "requirements", freeze_requireme
|
|
|
260
223
|
fp.writelines([ln + os.linesep for ln in requires] + [os.linesep])
|
|
261
224
|
|
|
262
225
|
|
|
263
|
-
def _retrieve_files(directory: str, *ext: str) ->
|
|
226
|
+
def _retrieve_files(directory: str, *ext: str) -> list[str]:
|
|
264
227
|
all_files = []
|
|
265
228
|
for root, _, files in os.walk(directory):
|
|
266
229
|
for fname in files:
|
|
@@ -270,7 +233,7 @@ def _retrieve_files(directory: str, *ext: str) -> List[str]:
|
|
|
270
233
|
return all_files
|
|
271
234
|
|
|
272
235
|
|
|
273
|
-
def _replace_imports(lines:
|
|
236
|
+
def _replace_imports(lines: list[str], mapping: list[tuple[str, str]], lightning_by: str = "") -> list[str]:
|
|
274
237
|
"""Replace imports of standalone package to lightning.
|
|
275
238
|
|
|
276
239
|
>>> lns = [
|
|
@@ -358,18 +321,20 @@ def copy_replace_imports(
|
|
|
358
321
|
fo.writelines(lines)
|
|
359
322
|
|
|
360
323
|
|
|
361
|
-
def create_mirror_package(source_dir: str, package_mapping:
|
|
324
|
+
def create_mirror_package(source_dir: str, package_mapping: dict[str, str]) -> None:
|
|
325
|
+
"""Create a mirror package with adjusted imports."""
|
|
362
326
|
# replace imports and copy the code
|
|
363
327
|
mapping = package_mapping.copy()
|
|
364
328
|
mapping.pop("lightning", None) # pop this key to avoid replacing `lightning` to `lightning.lightning`
|
|
365
329
|
|
|
366
330
|
mapping = {f"lightning.{sp}": sl for sp, sl in mapping.items()}
|
|
367
331
|
for pkg_from, pkg_to in mapping.items():
|
|
332
|
+
source_imports, target_imports = zip(*mapping.items())
|
|
368
333
|
copy_replace_imports(
|
|
369
334
|
source_dir=os.path.join(source_dir, pkg_from.replace(".", os.sep)),
|
|
370
335
|
# pytorch_lightning uses lightning_fabric, so we need to replace all imports for all directories
|
|
371
|
-
source_imports=
|
|
372
|
-
target_imports=
|
|
336
|
+
source_imports=source_imports,
|
|
337
|
+
target_imports=target_imports,
|
|
373
338
|
target_dir=os.path.join(source_dir, pkg_to.replace(".", os.sep)),
|
|
374
339
|
lightning_by=pkg_from,
|
|
375
340
|
)
|
|
@@ -397,7 +362,7 @@ class AssistantCLI:
|
|
|
397
362
|
if not ln_ or ln_.startswith("#"):
|
|
398
363
|
final.append(line)
|
|
399
364
|
continue
|
|
400
|
-
req = list(
|
|
365
|
+
req = list(_parse_requirements([ln_]))[0]
|
|
401
366
|
if req.name not in packages:
|
|
402
367
|
final.append(line)
|
|
403
368
|
print(final)
|
|
@@ -464,7 +429,7 @@ class AssistantCLI:
|
|
|
464
429
|
raise RuntimeError(f"Requesting file '{zip_url}' does not exist or it is just unavailable.")
|
|
465
430
|
|
|
466
431
|
with zipfile.ZipFile(zip_file, "r") as zip_ref:
|
|
467
|
-
zip_ref.extractall(tmp)
|
|
432
|
+
zip_ref.extractall(tmp)
|
|
468
433
|
|
|
469
434
|
zip_dirs = [d for d in glob.glob(os.path.join(tmp, "*")) if os.path.isdir(d)]
|
|
470
435
|
# check that the extracted archive has only repo folder
|
|
@@ -501,6 +466,21 @@ class AssistantCLI:
|
|
|
501
466
|
with open(rst_out, "w", encoding="utf-8") as fopen:
|
|
502
467
|
fopen.write(page)
|
|
503
468
|
|
|
469
|
+
@staticmethod
|
|
470
|
+
def convert_version2nightly(ver_file: str = "src/version.info") -> None:
|
|
471
|
+
"""Load the actual version and convert it to the nightly version."""
|
|
472
|
+
from datetime import datetime
|
|
473
|
+
|
|
474
|
+
with open(ver_file) as fo:
|
|
475
|
+
version = fo.read().strip()
|
|
476
|
+
# parse X.Y.Z version and prune any suffix
|
|
477
|
+
vers = re.match(r"(\d+)\.(\d+)\.(\d+).*", version)
|
|
478
|
+
# create timestamp YYYYMMDD
|
|
479
|
+
timestamp = datetime.now().strftime("%Y%m%d")
|
|
480
|
+
version = f"{'.'.join(vers.groups())}.dev{timestamp}"
|
|
481
|
+
with open(ver_file, "w") as fo:
|
|
482
|
+
fo.write(version + os.linesep)
|
|
483
|
+
|
|
504
484
|
|
|
505
485
|
if __name__ == "__main__":
|
|
506
486
|
import jsonargparse
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: lightning
|
|
3
|
-
Version: 2.
|
|
3
|
+
Version: 2.5.0.dev20241208
|
|
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.
|
|
@@ -21,66 +21,53 @@ Classifier: Topic :: Scientific/Engineering :: Information Analysis
|
|
|
21
21
|
Classifier: License :: OSI Approved :: Apache Software License
|
|
22
22
|
Classifier: Operating System :: OS Independent
|
|
23
23
|
Classifier: Programming Language :: Python :: 3
|
|
24
|
-
Classifier: Programming Language :: Python :: 3.8
|
|
25
24
|
Classifier: Programming Language :: Python :: 3.9
|
|
26
25
|
Classifier: Programming Language :: Python :: 3.10
|
|
27
26
|
Classifier: Programming Language :: Python :: 3.11
|
|
28
|
-
|
|
27
|
+
Classifier: Programming Language :: Python :: 3.12
|
|
28
|
+
Requires-Python: >=3.9
|
|
29
29
|
Description-Content-Type: text/markdown
|
|
30
|
-
Provides-Extra:
|
|
31
|
-
Provides-Extra:
|
|
32
|
-
Provides-Extra:
|
|
33
|
-
Provides-Extra: app-test
|
|
34
|
-
Provides-Extra: store-test
|
|
35
|
-
Provides-Extra: pytorch-strategies
|
|
30
|
+
Provides-Extra: fabric-test
|
|
31
|
+
Provides-Extra: fabric-strategies
|
|
32
|
+
Provides-Extra: fabric-examples
|
|
36
33
|
Provides-Extra: pytorch-test
|
|
34
|
+
Provides-Extra: pytorch-strategies
|
|
37
35
|
Provides-Extra: pytorch-extra
|
|
38
36
|
Provides-Extra: pytorch-examples
|
|
39
|
-
Provides-Extra: fabric-strategies
|
|
40
|
-
Provides-Extra: fabric-test
|
|
41
|
-
Provides-Extra: fabric-examples
|
|
42
37
|
Provides-Extra: fabric-all
|
|
43
38
|
Provides-Extra: fabric-dev
|
|
44
39
|
Provides-Extra: pytorch-all
|
|
45
40
|
Provides-Extra: pytorch-dev
|
|
46
|
-
Provides-Extra: app-extra
|
|
47
|
-
Provides-Extra: app-all
|
|
48
|
-
Provides-Extra: app-dev
|
|
49
|
-
Provides-Extra: components
|
|
50
|
-
Provides-Extra: cloud
|
|
51
|
-
Provides-Extra: ui
|
|
52
|
-
Provides-Extra: app
|
|
53
41
|
Provides-Extra: test
|
|
54
42
|
Provides-Extra: strategies
|
|
55
|
-
Provides-Extra: extra
|
|
56
43
|
Provides-Extra: examples
|
|
57
44
|
Provides-Extra: data
|
|
45
|
+
Provides-Extra: extra
|
|
58
46
|
Provides-Extra: all
|
|
59
47
|
Provides-Extra: dev
|
|
60
|
-
Provides-Extra: store
|
|
61
48
|
License-File: LICENSE
|
|
62
49
|
|
|
63
50
|
<div align="center">
|
|
64
51
|
|
|
65
|
-
<img alt="Lightning" src="https://pl-
|
|
52
|
+
<img alt="Lightning" src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/ptl_banner.png" width="800px" style="max-width: 100%;">
|
|
66
53
|
|
|
67
54
|
<br/>
|
|
68
55
|
<br/>
|
|
69
56
|
|
|
70
57
|
**The deep learning framework to pretrain, finetune and deploy AI models.**
|
|
71
58
|
|
|
72
|
-
**NEW-
|
|
59
|
+
**NEW- Deploying models? Check out [LitServe](https://github.com/Lightning-AI/litserve), the PyTorch Lightning for model serving**
|
|
73
60
|
|
|
74
61
|
______________________________________________________________________
|
|
75
62
|
|
|
76
63
|
<p align="center">
|
|
77
|
-
|
|
78
|
-
<a href="
|
|
79
|
-
<a href="
|
|
80
|
-
<a href="
|
|
81
|
-
<a href="https://
|
|
64
|
+
<a href="#quick-start" style="margin: 0 10px;">Quick start</a> •
|
|
65
|
+
<a href="#examples">Examples</a> •
|
|
66
|
+
<a href="#why-pytorch-lightning">PyTorch Lightning</a> •
|
|
67
|
+
<a href="#lightning-fabric-expert-control">Fabric</a> •
|
|
68
|
+
<a href="https://lightning.ai/">Lightning AI</a> •
|
|
82
69
|
<a href="#community">Community</a> •
|
|
83
|
-
<a href="https://lightning.
|
|
70
|
+
<a href="https://pytorch-lightning.readthedocs.io/en/2.5.0.dev20241208">Docs</a>
|
|
84
71
|
</p>
|
|
85
72
|
|
|
86
73
|
<!-- DO NOT ADD CONDA DOWNLOADS... README CHANGES MUST BE APPROVED BY EDEN OR WILL -->
|
|
@@ -101,25 +88,27 @@ ______________________________________________________________________
|
|
|
101
88
|
|
|
102
89
|
</div>
|
|
103
90
|
|
|
104
|
-
|
|
91
|
+
<div align="center">
|
|
92
|
+
|
|
93
|
+
<p align="center">
|
|
105
94
|
|
|
106
|
-
|
|
95
|
+
|
|
96
|
+
|
|
97
|
+
<a target="_blank" href="https://lightning.ai/docs/pytorch/latest/starter/introduction.html#define-a-lightningmodule">
|
|
98
|
+
<img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/get-started-badge.svg" height="36px" alt="Get started"/>
|
|
99
|
+
</a>
|
|
107
100
|
|
|
108
|
-
|
|
109
|
-
pip install lightning
|
|
110
|
-
```
|
|
101
|
+
</p>
|
|
111
102
|
|
|
112
|
-
|
|
103
|
+
</div>
|
|
113
104
|
|
|
114
|
-
|
|
105
|
+
|
|
115
106
|
|
|
116
|
-
|
|
107
|
+
# Lightning has 2 core packages
|
|
117
108
|
|
|
118
|
-
[PyTorch Lightning: Train and deploy PyTorch at scale](#pytorch-lightning
|
|
109
|
+
[PyTorch Lightning: Train and deploy PyTorch at scale](#why-pytorch-lightning).
|
|
119
110
|
<br/>
|
|
120
111
|
[Lightning Fabric: Expert control](#lightning-fabric-expert-control).
|
|
121
|
-
<br/>
|
|
122
|
-
[Lightning Apps: Build AI products and ML workflows](#lightning-apps-build-ai-products-and-ml-workflows).
|
|
123
112
|
|
|
124
113
|
Lightning gives you granular control over how much abstraction you want to add over PyTorch.
|
|
125
114
|
|
|
@@ -127,17 +116,19 @@ Lightning gives you granular control over how much abstraction you want to add o
|
|
|
127
116
|
<img src="https://pl-public-data.s3.amazonaws.com/assets_lightning/continuum.png" width="80%">
|
|
128
117
|
</div>
|
|
129
118
|
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
# PyTorch Lightning: Train and Deploy PyTorch at Scale
|
|
133
|
-
|
|
134
|
-
PyTorch Lightning is just organized PyTorch - Lightning disentangles PyTorch code to decouple the science from the engineering.
|
|
119
|
+
|
|
135
120
|
|
|
121
|
+
# Quick start
|
|
122
|
+
Install Lightning:
|
|
136
123
|
|
|
124
|
+
```bash
|
|
125
|
+
pip install lightning
|
|
126
|
+
```
|
|
137
127
|
|
|
138
|
-
|
|
128
|
+
<!-- -->
|
|
139
129
|
|
|
140
|
-
###
|
|
130
|
+
### PyTorch Lightning example
|
|
131
|
+
Define the training workflow. Here's a toy example ([explore real examples](https://lightning.ai/lightning-ai/studios?view=public§ion=featured&query=pytorch+lightning)):
|
|
141
132
|
|
|
142
133
|
```python
|
|
143
134
|
# main.py
|
|
@@ -165,7 +156,7 @@ class LitAutoEncoder(L.LightningModule):
|
|
|
165
156
|
|
|
166
157
|
def training_step(self, batch, batch_idx):
|
|
167
158
|
# training_step defines the train loop. It is independent of forward
|
|
168
|
-
x,
|
|
159
|
+
x, _ = batch
|
|
169
160
|
x = x.view(x.size(0), -1)
|
|
170
161
|
z = self.encoder(x)
|
|
171
162
|
x_hat = self.decoder(z)
|
|
@@ -199,6 +190,36 @@ pip install torchvision
|
|
|
199
190
|
python main.py
|
|
200
191
|
```
|
|
201
192
|
|
|
193
|
+
|
|
194
|
+
|
|
195
|
+
|
|
196
|
+
# Why PyTorch Lightning?
|
|
197
|
+
|
|
198
|
+
PyTorch Lightning is just organized PyTorch - Lightning disentangles PyTorch code to decouple the science from the engineering.
|
|
199
|
+
|
|
200
|
+
|
|
201
|
+
|
|
202
|
+
|
|
203
|
+
|
|
204
|
+
----
|
|
205
|
+
|
|
206
|
+
### Examples
|
|
207
|
+
Explore various types of training possible with PyTorch Lightning. Pretrain and finetune ANY kind of model to perform ANY task like classification, segmentation, summarization and more:
|
|
208
|
+
|
|
209
|
+
| Task | Description | Run |
|
|
210
|
+
|-------------------------------------------------------------------------------------------------------------|----------------------------------------------------------------|---|
|
|
211
|
+
| [Hello world](#hello-simple-model) | Pretrain - Hello world example | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/pytorch-lightning-hello-world"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
212
|
+
| [Image classification](https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning) | Finetune - ResNet-34 model to classify images of cars | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-classification-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
213
|
+
| [Image segmentation](https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning) | Finetune - ResNet-50 model to segment images | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/image-segmentation-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
214
|
+
| [Object detection](https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning) | Finetune - Faster R-CNN model to detect objects | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/object-detection-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
215
|
+
| [Text classification](https://lightning.ai/lightning-ai/studios/text-classification-with-pytorch-lightning) | Finetune - text classifier (BERT model) | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/text-classification-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
216
|
+
| [Text summarization](https://lightning.ai/lightning-ai/studios/text-summarization-with-pytorch-lightning) | Finetune - text summarization (Hugging Face transformer model) | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/text-summarization-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
217
|
+
| [Audio generation](https://lightning.ai/lightning-ai/studios/finetune-a-personal-ai-music-generator) | Finetune - audio generator (transformer model) | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/finetune-a-personal-ai-music-generator"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
218
|
+
| [LLM finetuning](https://lightning.ai/lightning-ai/studios/finetune-an-llm-with-pytorch-lightning) | Finetune - LLM (Meta Llama 3.1 8B) | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/finetune-an-llm-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
219
|
+
| [Image generation](https://lightning.ai/lightning-ai/studios/train-a-diffusion-model-with-pytorch-lightning) | Pretrain - Image generator (diffusion model) | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/train-a-diffusion-model-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
220
|
+
| [Recommendation system](https://lightning.ai/lightning-ai/studios/recommendation-system-with-pytorch-lightning) | Train - recommendation system (factorization and embedding) | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/recommendation-system-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
221
|
+
| [Time-series forecasting](https://lightning.ai/lightning-ai/studios/time-series-forecasting-with-pytorch-lightning) | Train - Time-series forecasting with LSTM | <a target="_blank" href="https://lightning.ai/lightning-ai/studios/time-series-forecasting-with-pytorch-lightning"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open In Studio"/></a> |
|
|
222
|
+
|
|
202
223
|
______________________________________________________________________
|
|
203
224
|
|
|
204
225
|
## Advanced features
|
|
@@ -338,7 +359,10 @@ ______________________________________________________________________
|
|
|
338
359
|
|
|
339
360
|
______________________________________________________________________
|
|
340
361
|
|
|
341
|
-
|
|
362
|
+
|
|
363
|
+
|
|
364
|
+
|
|
365
|
+
# Lightning Fabric: Expert control
|
|
342
366
|
|
|
343
367
|
Run on any device at any scale with expert-level control over PyTorch training loop and scaling strategy. You can even write your own Trainer.
|
|
344
368
|
|
|
@@ -520,62 +544,8 @@ ______________________________________________________________________
|
|
|
520
544
|
|
|
521
545
|
______________________________________________________________________
|
|
522
546
|
|
|
523
|
-
|
|
524
|
-
|
|
525
|
-
Lightning Apps remove the cloud infrastructure boilerplate so you can focus on solving the research or business problems. Lightning Apps can run on the Lightning Cloud, your own cluster or a private cloud.
|
|
526
|
-
|
|
527
|
-
<div align="center">
|
|
528
|
-
<img src="https://pl-public-data.s3.amazonaws.com/assets_lightning/lightning-apps-teaser.png" width="80%">
|
|
529
|
-
</div>
|
|
530
|
-
|
|
531
|
-
## Hello Lightning app world
|
|
532
|
-
|
|
533
|
-
```python
|
|
534
|
-
# app.py
|
|
535
|
-
import lightning as L
|
|
536
|
-
|
|
537
|
-
|
|
538
|
-
class TrainComponent(L.LightningWork):
|
|
539
|
-
def run(self, x):
|
|
540
|
-
print(f"train a model on {x}")
|
|
541
|
-
|
|
542
|
-
|
|
543
|
-
class AnalyzeComponent(L.LightningWork):
|
|
544
|
-
def run(self, x):
|
|
545
|
-
print(f"analyze model on {x}")
|
|
546
|
-
|
|
547
|
-
|
|
548
|
-
class WorkflowOrchestrator(L.LightningFlow):
|
|
549
|
-
def __init__(self) -> None:
|
|
550
|
-
super().__init__()
|
|
551
|
-
self.train = TrainComponent(cloud_compute=L.CloudCompute("cpu"))
|
|
552
|
-
self.analyze = AnalyzeComponent(cloud_compute=L.CloudCompute("gpu"))
|
|
553
|
-
|
|
554
|
-
def run(self):
|
|
555
|
-
self.train.run("CPU machine 1")
|
|
556
|
-
self.analyze.run("GPU machine 2")
|
|
557
|
-
|
|
558
|
-
|
|
559
|
-
app = L.LightningApp(WorkflowOrchestrator())
|
|
560
|
-
```
|
|
561
|
-
|
|
562
|
-
Run on the cloud or locally
|
|
563
|
-
|
|
564
|
-
```bash
|
|
565
|
-
# run on the cloud
|
|
566
|
-
lightning run app app.py --setup --cloud
|
|
567
|
-
|
|
568
|
-
# run locally
|
|
569
|
-
lightning run app app.py
|
|
570
|
-
```
|
|
571
|
-
|
|
572
|
-
______________________________________________________________________
|
|
573
|
-
|
|
574
|
-
<div align="center">
|
|
575
|
-
<a href="https://lightning.ai/docs/app/stable/">Read the Lightning Apps docs</a>
|
|
576
|
-
</div>
|
|
577
|
-
|
|
578
|
-
______________________________________________________________________
|
|
547
|
+
|
|
548
|
+
|
|
579
549
|
|
|
580
550
|
## Examples
|
|
581
551
|
|
|
@@ -606,7 +576,8 @@ ______________________________________________________________________
|
|
|
606
576
|
- [Logistic Regression](https://lightning-bolts.readthedocs.io/en/stable/models/classic_ml.html#logistic-regression)
|
|
607
577
|
- [Linear Regression](https://lightning-bolts.readthedocs.io/en/stable/models/classic_ml.html#linear-regression)
|
|
608
578
|
|
|
609
|
-
|
|
579
|
+
|
|
580
|
+
|
|
610
581
|
|
|
611
582
|
## Continuous Integration
|
|
612
583
|
|
|
@@ -621,8 +592,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
|
|
|
621
592
|
|
|
622
593
|
| System / PyTorch ver. | 1.13 | 2.0 | 2.1 |
|
|
623
594
|
| :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
|
|
624
|
-
| Linux py3.9 \[GPUs\] | | | [](https://github.com/Lightning-AI/lightning/actions/workflows/tpu-tests.yml) | |
|
|
595
|
+
| Linux py3.9 \[GPUs\] | | | [](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
|
|
626
596
|
| Linux (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
627
597
|
| OSX (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
628
598
|
| Windows (multiple Python versions) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) | [](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |
|
|
@@ -630,7 +600,8 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
|
|
|
630
600
|
</center>
|
|
631
601
|
</details>
|
|
632
602
|
|
|
633
|
-
|
|
603
|
+
|
|
604
|
+
|
|
634
605
|
|
|
635
606
|
## Community
|
|
636
607
|
|