lightning 2.4.0.dev20240630__tar.gz → 2.4.0.dev20240707__tar.gz

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/.actions/assistant.py

@@ -14,10 +14,8 @@
 import glob
 import logging
 import os
-import pathlib
 import re
 import shutil
-import tarfile
 import tempfile
 import urllib.request
 from distutils.version import LooseVersion
@@ -35,11 +33,6 @@ REQUIREMENT_FILES = {
         "requirements/pytorch/strategies.txt",
         "requirements/pytorch/examples.txt",
     ),
-    "app": (
-        "requirements/app/app.txt",
-        "requirements/app/cloud.txt",
-        "requirements/app/ui.txt",
-    ),
     "fabric": (
         "requirements/fabric/base.txt",
         "requirements/fabric/strategies.txt",
@@ -216,30 +209,6 @@ def distribute_version(src_folder: str, ver_file: str = "version.info") -> None:
         shutil.copy2(ver_template, fpath)
 
 
-def _download_frontend(pkg_path: str, version: str = "v0.0.0"):
-    """Downloads an archive file for a specific release of the Lightning frontend and extracts it to the correct
-    directory."""
-
-    try:
-        frontend_dir = pathlib.Path(pkg_path, "ui")
-        download_dir = tempfile.mkdtemp()
-
-        shutil.rmtree(frontend_dir, ignore_errors=True)
-        # TODO: remove this once lightning-ui package is ready as a dependency
-        frontend_release_url = f"https://lightning-packages.s3.amazonaws.com/ui/{version}.tar.gz"
-        response = urllib.request.urlopen(frontend_release_url)
-
-        file = tarfile.open(fileobj=response, mode="r|gz")
-        file.extractall(path=download_dir)  # noqa: S202
-
-        shutil.move(download_dir, frontend_dir)
-        print("The Lightning UI has successfully been downloaded!")
-
-    # If installing from source without internet connection, we don't want to break the installation
-    except Exception:
-        print("The Lightning UI downloading has failed!")
-
-
 def _load_aggregate_requirements(req_dir: str = "requirements", freeze_requirements: bool = False) -> None:
     """Load all base requirements from all particular packages and prune duplicates.
 
@@ -466,7 +435,7 @@ class AssistantCLI:
                 raise RuntimeError(f"Requesting file '{zip_url}' does not exist or it is just unavailable.")
 
             with zipfile.ZipFile(zip_file, "r") as zip_ref:
-                zip_ref.extractall(tmp)  # noqa: S202
+                zip_ref.extractall(tmp)
 
             zip_dirs = [d for d in glob.glob(os.path.join(tmp, "*")) if os.path.isdir(d)]
             # check that the extracted archive has only repo folder

{lightning-2.4.0.dev20240630/src/lightning.egg-info → lightning-2.4.0.dev20240707}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning
-Version: 2.4.0.dev20240630
+Version: 2.4.0.dev20240707
 Summary: The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
 Home-page: https://github.com/Lightning-AI/lightning
 Author: Lightning AI et al.
@@ -27,11 +27,6 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Provides-Extra: app-ui
-Provides-Extra: app-components
-Provides-Extra: app-cloud
-Provides-Extra: app-test
-Provides-Extra: store-test
 Provides-Extra: fabric-strategies
 Provides-Extra: fabric-examples
 Provides-Extra: fabric-test
@@ -43,21 +38,13 @@ Provides-Extra: fabric-all
 Provides-Extra: fabric-dev
 Provides-Extra: pytorch-all
 Provides-Extra: pytorch-dev
-Provides-Extra: app-extra
-Provides-Extra: app-all
-Provides-Extra: app-dev
-Provides-Extra: ui
-Provides-Extra: components
-Provides-Extra: cloud
-Provides-Extra: test
-Provides-Extra: app
 Provides-Extra: data
 Provides-Extra: strategies
 Provides-Extra: examples
+Provides-Extra: test
 Provides-Extra: extra
 Provides-Extra: all
 Provides-Extra: dev
-Provides-Extra: store
 License-File: LICENSE
 
 <div align="center">
@@ -78,7 +65,7 @@ ______________________________________________________________________
   <a href="#examples">Examples</a> •
   <a href="#pytorch-lightning-train-and-deploy-pytorch-at-scale">PyTorch Lightning</a> •
   <a href="#lightning-fabric-expert-control">Fabric</a> •
-  <a href="https://pytorch-lightning.readthedocs.io/en/2.4.0.dev20240630">Docs</a> •
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.4.0.dev20240707">Docs</a> •
   <a href="#community">Community</a> •
   <a href="https://lightning.ai/docs/pytorch/stable/generated/CONTRIBUTING.html">Contribute</a> •
 </p>
@@ -154,11 +141,12 @@ ______________________________________________________________________
 ### Examples
 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:    
 
-| Task  | Description | Run |
-|---|---|---|
-| [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> |
-| [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> |   
-| [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> |   
+| Task                                                                                                        | Description                                    | Run |
+|-------------------------------------------------------------------------------------------------------------|------------------------------------------------|---|
+| [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> |
+| [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> |   
+| [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> |   
+| [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> |   
 
 ### Hello simple model
 
@@ -594,7 +582,7 @@ Lightning is rigorously tested across multiple CPUs, GPUs and TPUs and against m
 
 |       System / PyTorch ver.        | 1.13                                                                                                                                                                                                                            | 2.0                                                                                                                                                                                                                             |                                                                                                               2.1                                                                                                               |
 | :--------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|
-|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.4.0.dev20240630)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
+|        Linux py3.9 \[GPUs\]        |  |  | [![Build Status](https://dev.azure.com/Lightning-AI/lightning/_apis/build/status%2Fpytorch-lightning%20%28GPUs%29?branchName=refs%2Ftags%2F2.4.0.dev20240707)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) |
 |        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)     |      |
 |  Linux (multiple Python versions)  | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |
 |   OSX (multiple Python versions)   | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 |

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/README.md

@@ -133,11 +133,12 @@ ______________________________________________________________________
 ### Examples
 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:    
 
-| Task  | Description | Run |
-|---|---|---|
-| [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> |
-| [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> |   
-| [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> |   
+| Task                                                                                                        | Description                                    | Run |
+|-------------------------------------------------------------------------------------------------------------|------------------------------------------------|---|
+| [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> |
+| [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> |   
+| [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> |   
+| [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> |   
 
 ### Hello simple model
 

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/pyproject.toml

@@ -134,14 +134,7 @@ max-complexity = 10
 files = [
     "src/lightning",
 ]
-# This section is for folders with "-" as they are not valid python modules
-exclude = [
-    "src/lightning/app/cli/app-template",
-    "src/lightning/app/cli/component-template",
-    "src/lightning/app/cli/pl-app-template",
-    "src/lightning/app/cli/react-ui-template",
-    "src/lightning/app/launcher/utils.py",
-]
+
 install_types = "True"
 non_interactive = "True"
 disallow_untyped_defs = "True"
@@ -156,100 +149,6 @@ disable_error_code = "attr-defined"
 # style choices
 warn_no_return = "False"
 
-# Ignore mypy errors for these files
-# TODO: the goal is for this to be empty
-[[tool.mypy.overrides]]
-# the list can be generated with:
-# mypy --no-error-summary 2>&1 | tr ':' ' ' | awk '{print $1}' | sort | uniq | sed 's/\.py//g; s|src/||g;  s|\/|\.|g' | xargs -I {} echo '"{}",'
-module = [
-    "lightning.app.api.http_methods",
-    "lightning.app.api.request_types",
-    "lightning.app.cli.cmd_install",
-    "lightning.app.cli.commands.app_commands",
-    "lightning.app.cli.commands.cd",
-    "lightning.app.cli.commands.cp",
-    "lightning.app.cli.commands.ls",
-    "lightning.app.cli.connect.app",
-    "lightning.app.components.database.client",
-    "lightning.app.components.database.server",
-    "lightning.app.components.database.utilities",
-    "lightning.app.components.multi_node.base",
-    "lightning.app.components.multi_node.fabric",
-    "lightning.app.components.multi_node.pytorch_spawn",
-    "lightning.app.components.multi_node.trainer",
-    "lightning.app.components.python.popen",
-    "lightning.app.components.python.tracer",
-    "lightning.app.components.serve.auto_scaler",
-    "lightning.app.components.serve.gradio_server",
-    "lightning.app.components.serve.python_server",
-    "lightning.app.components.serve.serve",
-    "lightning.app.components.serve.streamlit",
-    "lightning.app.components.serve.types.image",
-    "lightning.app.components.serve.types.type",
-    "lightning.app.components.training",
-    "lightning.app.frontend.panel.app_state_comm",
-    "lightning.app.frontend.panel.app_state_watcher",
-    "lightning.app.frontend.panel.panel_frontend",
-    "lightning.app.frontend.panel.panel_serve_render_fn",
-    "lightning.app.frontend.streamlit_base",
-    "lightning.app.frontend.stream_lit",
-    "lightning.app.frontend.utils",
-    "lightning.app.frontend.web",
-    "lightning.app.launcher.launcher",
-    "lightning.app.launcher.lightning_backend",
-    "lightning.app.launcher.lightning_hybrid_backend",
-    "lightning.app.pdb.pdb",
-    "lightning.app.runners.backends.backend",
-    "lightning.app.runners.backends.cloud",
-    "lightning.app.runners.backends.docker",
-    "lightning.app.runners.backends.mp_process",
-    "lightning.app.runners.cloud",
-    "lightning.app.runners.multiprocess",
-    "lightning.app.runners.runtime",
-    "lightning.app.source_code.copytree",
-    "lightning.app.source_code.hashing",
-    "lightning.app.source_code.local",
-    "lightning.app.source_code.tar",
-    "lightning.app.source_code.uploader",
-    "lightning.app.storage.copier",
-    "lightning.app.storage.drive",
-    "lightning.app.storage.filesystem",
-    "lightning.app.storage.orchestrator",
-    "lightning.app.storage.path",
-    "lightning.app.storage.payload",
-    "lightning.app.structures.dict",
-    "lightning.app.structures.list",
-    "lightning.app.testing.helpers",
-    "lightning.app.testing.testing",
-    "lightning.app.utilities.app_helpers",
-    "lightning.app.utilities.app_logs",
-    "lightning.app.utilities.cli_helpers",
-    "lightning.app.utilities.cloud",
-    "lightning.app.utilities.commands.base",
-    "lightning.app.utilities.component",
-    "lightning.app.utilities.enum",
-    "lightning.app.utilities.exceptions",
-    "lightning.app.utilities.git",
-    "lightning.app.utilities.imports",
-    "lightning.app.utilities.introspection",
-    "lightning.app.utilities.layout",
-    "lightning.app.utilities.load_app",
-    "lightning.app.utilities.log_helpers",
-    "lightning.app.utilities.login",
-    "lightning.app.utilities.name_generator",
-    "lightning.app.utilities.network",
-    "lightning.app.utilities.openapi",
-    "lightning.app.utilities.packaging.cloud_compute",
-    "lightning.app.utilities.packaging.lightning_utils",
-    "lightning.app.utilities.proxies",
-    "lightning.app.utilities.scheduler",
-    "lightning.app.utilities.state",
-    "lightning.app.utilities.tracer",
-    "lightning.app.utilities.tree",
-    "lightning.store.utils",
-]
-ignore_errors = "True"
-
 
 [tool.coverage.report]
 exclude_lines = [

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/requirements/base.txt

@@ -1,9 +1,9 @@
 PyYAML<7.0,>=5.4
 fsspec[http]<2025.0,>=2022.5.0
 lightning-utilities<1.0,>=0.10.0
-numpy<2.0,>=1.17.2
+numpy<2.0,>=1.21.0
 packaging<24.0,>=20.0
-torch<3.0,>=2.0.0
+torch<3.0,>=2.1.0
 torchmetrics<2.0,>=0.7.0
 tqdm<5.0,>=4.57.0
 typing-extensions<5.0,>=4.4.0

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/requirements/fabric/base.txt

@@ -1,8 +1,8 @@
 # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
-numpy >=1.17.2, <1.27.0
-torch >=2.0.0, <2.4.0
+numpy >=1.21.0, <1.27.0
+torch >=2.1.0, <2.5.0
 fsspec[http] >=2022.5.0, <2024.4.0
 packaging >=20.0, <=23.1
 typing-extensions >=4.4.0, <4.10.0

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/requirements/fabric/examples.txt

@@ -1,6 +1,6 @@
 # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
-torchvision >=0.15.0, <0.19.0
+torchvision >=0.16.0, <0.20.0
 torchmetrics >=0.10.0, <1.3.0
 lightning-utilities >=0.8.0, <0.12.0

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/requirements/pytorch/base.txt

@@ -1,8 +1,8 @@
 # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
-numpy >=1.17.2, <1.27.0
-torch >=2.0.0, <2.4.0
+numpy >=1.21.0, <1.27.0
+torch >=2.1.0, <2.4.0
 tqdm >=4.57.0, <4.67.0
 PyYAML >=5.4, <6.1.0
 fsspec[http] >=2022.5.0, <2024.4.0

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/requirements/pytorch/examples.txt

@@ -2,7 +2,7 @@
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 
 requests <2.32.0
-torchvision >=0.15.0, <0.19.0
+torchvision >=0.16.0, <0.19.0
 ipython[all] <8.15.0
 torchmetrics >=0.10.0, <1.3.0
 lightning-utilities >=0.8.0, <0.12.0

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/requirements/pytorch/test.txt

@@ -8,8 +8,8 @@ pytest-random-order ==1.1.0
 # needed in tests
 cloudpickle >=1.3, <2.3.0
 scikit-learn >0.22.1, <1.4.0
-onnx >=0.14.0, <1.15.0
-onnxruntime >=0.15.0, <1.17.0
+onnx >=1.12.0, <1.15.0
+onnxruntime >=1.12.0, <1.17.0
 psutil <5.9.6 # for `DeviceStatsMonitor`
 pandas >1.0, <2.2.0  # needed in benchmarks
 fastapi  # for `ServableModuleValidator`  # not setting version as re-defined in App

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/setup.py

@@ -17,7 +17,7 @@ installation for all packages.
 
 There are considered three main scenarios for installing this project:
 
-1. Using PyPI registry when you can install `pytorch-lightning`, `lightning-app`, etc. or `lightning` for all.
+1. Using PyPI registry when you can install `pytorch-lightning`, etc. or `lightning` for all.
 
 2. Installation from source code after cloning repository.
     In such case we recommend to use command `pip install .` or `pip install -e .` for development version
@@ -26,12 +26,11 @@ There are considered three main scenarios for installing this project:
 
      - for `pytorch-lightning` use `export PACKAGE_NAME=pytorch ; pip install .`
      - for `lightning-fabric` use `export PACKAGE_NAME=fabric ; pip install .`
-     - for `lightning-app` use `export PACKAGE_NAME=app ; pip install .`
 
 3. Building packages as sdist or binary wheel and installing or publish to PyPI afterwords you use command
     `python setup.py sdist` or `python setup.py bdist_wheel` accordingly.
    In case you want to build just a particular package you want to set an environment variable:
-   `PACKAGE_NAME=lightning|pytorch|app|fabric python setup.py sdist|bdist_wheel`
+   `PACKAGE_NAME=lightning|pytorch|fabric python setup.py sdist|bdist_wheel`
 
 4. Automated releasing with GitHub action is natural extension of 3) is composed of three consecutive steps:
     a) determine which packages shall be released based on version increment in `__version__.py` and eventually
@@ -57,7 +56,6 @@ _PACKAGE_NAME = os.environ.get("PACKAGE_NAME")
 _PACKAGE_MAPPING = {
     "lightning": "lightning",
     "pytorch": "pytorch_lightning",
-    "app": "lightning_app",
     "fabric": "lightning_fabric",
 }
 # https://packaging.python.org/guides/single-sourcing-package-version/

lightning-2.4.0.dev20240630/src/lightning/1aadc1d656507543e9875427981f387a4b733ebc60cdada0 → lightning-2.4.0.dev20240707/src/lightning/694dbd34fcf659be5606c7f9f88c93d396c29747d6b13f33

@@ -2,11 +2,6 @@
 include *.cff  # citation info
 include .actions/assistant.py
 include requirements/base.txt
-include src/lightning/app/CHANGELOG.md
-include src/lightning/app/README.md
-include src/lightning/app/components/serve/catimage.png
-include src/lightning/app/py.typed  # marker file for PEP 561
-include src/lightning/app/version.info
 include src/lightning/fabric/CHANGELOG.md
 include src/lightning/fabric/README.md
 include src/lightning/fabric/py.typed  # marker file for PEP 561
@@ -18,8 +13,5 @@ include src/lightning/pytorch/py.typed  # marker file for PEP 561
 include src/lightning/pytorch/version.info
 include src/lightning/version.info
 include src/version.info
-recursive-include requirements/app *.txt
 recursive-include requirements/fabric *.txt
 recursive-include requirements/pytorch *.txt
-recursive-include src/lightning/app/cli/*-template *
-recursive-include src/lightning/app/ui *

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/__init__.py

@@ -1,7 +1,6 @@
 """Root package info."""
 
 import logging
-import sys
 
 # explicitly don't set root logger's propagation and leave this to subpackages to manage
 _logger = logging.getLogger(__name__)
@@ -31,19 +30,3 @@ __all__ = [
     "Fabric",
     "__version__",
 ]
-
-
-def _cli_entry_point() -> None:
-    from lightning_utilities.core.imports import ModuleAvailableCache, RequirementCache
-
-    if not (
-        ModuleAvailableCache("lightning.app")
-        if RequirementCache("lightning-utilities<0.10.0")
-        else RequirementCache(module="lightning.app")
-    ):
-        print("The `lightning` command requires additional dependencies: `pip install lightning[app]`")
-        sys.exit(1)
-
-    from lightning.app.cli.lightning_cli import main
-
-    main()

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/__setup__.py

@@ -45,12 +45,8 @@ def _prepare_extras() -> Dict[str, Any]:
     extras["fabric-dev"] = extras["fabric-all"] + extras["fabric-test"]
     extras["pytorch-all"] = extras["pytorch-extra"] + extras["pytorch-strategies"] + extras["pytorch-examples"]
     extras["pytorch-dev"] = extras["pytorch-all"] + extras["pytorch-test"]
-    extras["app-extra"] = extras["app-app"] + extras["app-cloud"] + extras["app-ui"] + extras["app-components"]
-    extras["app-all"] = extras["app-extra"]
-    extras["app-dev"] = extras["app-all"] + extras["app-test"]
-    extras["store-store"] = extras["app-app"]  # todo: consider cutting/leaning this dependency
 
-    # merge per-project extras of the same category, e.g. `app-test` + `fabric-test`
+    # merge per-project extras of the same category
     for extra in list(extras):
         name = "-".join(extra.split("-")[1:])
         extras[name] = extras.get(name, []) + extras[extra]
@@ -74,17 +70,6 @@ def _setup_args() -> Dict[str, Any]:
         _PROJECT_ROOT, homepage=about.__homepage__, version=version.version
     )
 
-    # TODO: remove this once lightning-ui package is ready as a dependency
-    ui_ver_file = os.path.join(_SOURCE_ROOT, "app-ui-version.info")
-    if os.path.isfile(ui_ver_file):
-        with open(ui_ver_file, encoding="utf-8") as fo:
-            ui_version = fo.readlines()[0].strip()
-        download_fe_version = {"version": ui_version}
-    else:
-        print(f"Missing file with FE version: {ui_ver_file}")
-        download_fe_version = {}
-    _ASSISTANT._download_frontend(os.path.join(_PACKAGE_ROOT, "app"), **download_fe_version)
-
     # TODO: consider invaliding some additional arguments from packages, for example if include data or safe to zip
 
     install_requires = _ASSISTANT.load_requirements(
@@ -114,7 +99,6 @@ def _setup_args() -> Dict[str, Any]:
             "console_scripts": [
                 "fabric = lightning.fabric.cli:_main",
                 "lightning = lightning.fabric.cli:_legacy_main",
-                "lightning_app = lightning:_cli_entry_point",
             ],
         },
         "setup_requires": [],

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/fabric/CHANGELOG.md

@@ -27,7 +27,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ### Removed
 
--
+- Removed support for PyTorch 2.1 ([#20009](https://github.com/Lightning-AI/lightning/pull/20009))
+
 
 -
 

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/fabric/__init__.py

@@ -21,7 +21,7 @@ if not _root_logger.hasHandlers():
     _logger.propagate = False
 
 
-# In PyTorch 2.0+, setting this variable will force `torch.cuda.is_available()` and `torch.cuda.device_count()`
+# Setting this variable will force `torch.cuda.is_available()` and `torch.cuda.device_count()`
 # to use an NVML-based implementation that doesn't poison forks.
 # https://github.com/pytorch/pytorch/issues/83973
 os.environ["PYTORCH_NVML_BASED_CUDA_CHECK"] = "1"

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/fabric/cli.py

@@ -57,18 +57,12 @@ if _CLICK_AVAILABLE:
 
         """
         hparams = sys.argv[1:]
-        if len(hparams) >= 2 and hparams[0] == "run":
-            if hparams[1] == "model":
-                print(
-                    "`lightning run model` is deprecated and will be removed in future versions."
-                    " Please call `fabric run` instead."
-                )
-                _main()
-                return
-
-            from lightning.app.cli.lightning_cli import main as main_cli
-
-            main_cli()
+        if len(hparams) >= 2 and hparams[0] == "run" and hparams[1] == "model":
+            print(
+                "`lightning run model` is deprecated and will be removed in future versions."
+                " Please call `fabric run` instead."
+            )
+            _main()
             return
 
         if _LIGHTNING_SDK_AVAILABLE:

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/fabric/fabric.py

@@ -225,8 +225,7 @@ class Fabric:
             _reapply_compile: If ``True`` (default), and the model was ``torch.compile``d before, the
                 corresponding :class:`~torch._dynamo.OptimizedModule` wrapper will be removed and reapplied with the
                 same settings after the model was set up by the strategy (e.g., after the model was wrapped by DDP,
-                FSDP etc.). Only applies on PyTorch >= 2.1. Set it to ``False`` if compiling DDP/FSDP is causing
-                issues.
+                FSDP etc.). Set it to ``False`` if compiling DDP/FSDP is causing issues.
 
         Returns:
             The tuple containing wrapped module and the optimizers, in the same order they were passed in.
@@ -292,8 +291,7 @@ class Fabric:
             _reapply_compile: If ``True`` (default), and the model was ``torch.compile``d before, the
                 corresponding :class:`~torch._dynamo.OptimizedModule` wrapper will be removed and reapplied with the
                 same settings after the model was set up by the strategy (e.g., after the model was wrapped by DDP,
-                FSDP etc.). Only applies on PyTorch >= 2.1. Set it to ``False`` if compiling DDP/FSDP is causing
-                issues.
+                FSDP etc.). Set it to ``False`` if compiling DDP/FSDP is causing issues.
         Returns:
             The wrapped model.
 

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/fabric/loggers/tensorboard.py

@@ -107,7 +107,7 @@ class TensorBoardLogger(Logger):
         self._prefix = prefix
         self._fs = get_filesystem(root_dir)
 
-        self._experiment: Optional["SummaryWriter"] = None
+        self._experiment: Optional[SummaryWriter] = None
         self._kwargs = kwargs
 
     @property

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/fabric/plugins/precision/amp.py

@@ -22,6 +22,7 @@ from typing_extensions import override
 
 from lightning.fabric.plugins.precision.precision import Precision
 from lightning.fabric.plugins.precision.utils import _convert_fp_tensor
+from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_4
 from lightning.fabric.utilities.types import Optimizable
 
 
@@ -39,7 +40,7 @@ class MixedPrecision(Precision):
         self,
         precision: Literal["16-mixed", "bf16-mixed"],
         device: str,
-        scaler: Optional[torch.cuda.amp.GradScaler] = None,
+        scaler: Optional["torch.cuda.amp.GradScaler"] = None,
     ) -> None:
         if precision not in ("16-mixed", "bf16-mixed"):
             raise ValueError(
@@ -49,7 +50,7 @@ class MixedPrecision(Precision):
 
         self.precision = precision
         if scaler is None and self.precision == "16-mixed":
-            scaler = torch.cuda.amp.GradScaler()
+            scaler = torch.amp.GradScaler(device=device) if _TORCH_GREATER_EQUAL_2_4 else torch.cuda.amp.GradScaler()
         if scaler is not None and self.precision == "bf16-mixed":
             raise ValueError(f"`precision='bf16-mixed'` does not use a scaler, found {scaler}.")
         self.device = device

{lightning-2.4.0.dev20240630 → lightning-2.4.0.dev20240707}/src/lightning/fabric/strategies/deepspeed.py

@@ -291,7 +291,7 @@ class DeepSpeedStrategy(DDPStrategy, _Sharded):
         self.hysteresis = hysteresis
         self.min_loss_scale = min_loss_scale
 
-        self._deepspeed_engine: Optional["DeepSpeedEngine"] = None
+        self._deepspeed_engine: Optional[DeepSpeedEngine] = None
 
     @property
     def zero_stage_3(self) -> bool: