lightning 2.3.2__tar.gz → 2.3.3__tar.gz

{lightning-2.3.2 → lightning-2.3.3}/.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.3.2/src/lightning.egg-info → lightning-2.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning
-Version: 2.3.2
+Version: 2.3.3
 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,10 +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
@@ -43,21 +39,14 @@ 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: test
 Provides-Extra: data
 Provides-Extra: strategies
 Provides-Extra: examples
 Provides-Extra: extra
 Provides-Extra: all
 Provides-Extra: dev
-Provides-Extra: store
 License-File: LICENSE
 
 <div align="center">
@@ -78,7 +67,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.3.2">Docs</a> •
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.3.3">Docs</a> •
   <a href="#community">Community</a> •
   <a href="https://lightning.ai/docs/pytorch/stable/generated/CONTRIBUTING.html">Contribute</a> •
 </p>
@@ -594,7 +583,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.3.2)](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.3.3)](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.3.2 → lightning-2.3.3}/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.3.2 → lightning-2.3.3}/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.3.2 → lightning-2.3.3}/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.3.2 → lightning-2.3.3}/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(

lightning-2.3.2/src/lightning/6f44f694807e415826050f8cb64a3c1736c7be298280703f → lightning-2.3.3/src/lightning/af9f07bb212d5f502afa93ec4c33977c308c6110fb2e6081

@@ -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.3.3/src/lightning/app/__init__.py

@@ -0,0 +1,26 @@
+import sys
+
+from lightning_utilities.core.imports import RequirementCache, module_available
+
+__all__ = []
+
+if not RequirementCache("lightning_app"):
+    raise ModuleNotFoundError("Please, run `pip install lightning-app`")  # E111
+
+else:
+    import lightning_app
+
+    # Enable resolution at least for lower data namespace
+    sys.modules["lightning.app"] = lightning_app
+
+    from lightning_app.core.app import LightningApp
+    from lightning_app.core.flow import LightningFlow
+    from lightning_app.core.work import LightningWork
+    from lightning_app.plugin.plugin import LightningPlugin
+    from lightning_app.utilities.packaging.build_config import BuildConfig
+    from lightning_app.utilities.packaging.cloud_compute import CloudCompute
+
+    if module_available("lightning_app.components.demo"):
+        from lightning.app.components import demo  # noqa: F401
+
+    __all__ = ["LightningApp", "LightningFlow", "LightningWork", "LightningPlugin", "BuildConfig", "CloudCompute"]

{lightning-2.3.2 → lightning-2.3.3}/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.3.3/src/lightning/version.info

@@ -0,0 +1 @@
+2.3.3

{lightning-2.3.2 → lightning-2.3.3/src/lightning.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning
-Version: 2.3.2
+Version: 2.3.3
 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,10 +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
@@ -43,21 +39,14 @@ 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: test
 Provides-Extra: data
 Provides-Extra: strategies
 Provides-Extra: examples
 Provides-Extra: extra
 Provides-Extra: all
 Provides-Extra: dev
-Provides-Extra: store
 License-File: LICENSE
 
 <div align="center">
@@ -78,7 +67,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.3.2">Docs</a> •
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.3.3">Docs</a> •
   <a href="#community">Community</a> •
   <a href="https://lightning.ai/docs/pytorch/stable/generated/CONTRIBUTING.html">Contribute</a> •
 </p>
@@ -594,7 +583,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.3.2)](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.3.3)](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)                 |