PyPI - lightning - Versions diffs - 2.0.3__tar.gz → 2.0.4__tar.gz - Mend

lightning 2.0.3tar.gz → 2.0.4tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (570) hide show

{lightning-2.0.3 → lightning-2.0.4}/.actions/assistant.py RENAMED Viewed

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import glob
+import logging
 import os
 import pathlib
 import re
@@ -142,7 +143,7 @@ def load_requirements(path_dir: str, file_name: str = "base.txt", unfreeze: str
     >>> path_req = os.path.join(_PROJECT_ROOT, "requirements")
     >>> load_requirements(path_req, "docs.txt", unfreeze="major")  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
-    ['sphinx<6.0,>=4.0', ...]
+    ['sphinx<...]
     """
     assert unfreeze in {"none", "major", "all"}
     path = Path(path_dir) / file_name
@@ -421,6 +422,42 @@ class AssistantCLI:
             source_dir, source_imports, target_imports, target_dir=target_dir, lightning_by=lightning_by
         )
+    @staticmethod
+    def pull_docs_files(
+        gh_user_repo: str,
+        target_dir: str = "docs/source-pytorch/XXX",
+        checkout: str = "tags/1.0.0",
+        source_dir: str = "docs/source",
+    ) -> None:
+        """Pull docs pages from external source and append to local docs."""
+        import zipfile
+        zip_url = f"https://github.com/{gh_user_repo}/archive/refs/{checkout}.zip"
+        with tempfile.TemporaryDirectory() as tmp:
+            zip_file = os.path.join(tmp, "repo.zip")
+            urllib.request.urlretrieve(zip_url, zip_file)
+            with zipfile.ZipFile(zip_file, "r") as zip_ref:
+                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
+            assert len(zip_dirs) == 1
+            repo_dir = zip_dirs[0]
+            ls_pages = glob.glob(os.path.join(repo_dir, source_dir, "*.rst"))
+            ls_pages += glob.glob(os.path.join(repo_dir, source_dir, "**", "*.rst"))
+            for rst in ls_pages:
+                rel_rst = rst.replace(os.path.join(repo_dir, source_dir) + os.path.sep, "")
+                rel_dir = os.path.dirname(rel_rst)
+                os.makedirs(os.path.join(_PROJECT_ROOT, target_dir, rel_dir), exist_ok=True)
+                new_rst = os.path.join(_PROJECT_ROOT, target_dir, rel_rst)
+                if os.path.isfile(new_rst):
+                    logging.warning(f"Page {new_rst} already exists in the local tree so it will be skipped.")
+                    continue
+                shutil.copy(rst, new_rst)
 if __name__ == "__main__":
     import jsonargparse

{lightning-2.0.3/src/lightning.egg-info → lightning-2.0.4}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning
-Version: 2.0.3
+Version: 2.0.4
 Summary: Use Lightning Apps to build everything from production-ready, multi-cloud ML systems to simple research demos.
 Home-page: https://github.com/Lightning-AI/lightning
 Author: Lightning AI et al.
@@ -26,17 +26,17 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
+Provides-Extra: pytorch-test
 Provides-Extra: pytorch-strategies
-Provides-Extra: pytorch-examples
 Provides-Extra: pytorch-extra
-Provides-Extra: pytorch-test
-Provides-Extra: fabric-strategies
-Provides-Extra: fabric-examples
-Provides-Extra: fabric-test
+Provides-Extra: pytorch-examples
+Provides-Extra: app-test
+Provides-Extra: app-cloud
 Provides-Extra: app-ui
 Provides-Extra: app-components
-Provides-Extra: app-cloud
-Provides-Extra: app-test
+Provides-Extra: fabric-test
+Provides-Extra: fabric-strategies
+Provides-Extra: fabric-examples
 Provides-Extra: fabric-all
 Provides-Extra: fabric-dev
 Provides-Extra: pytorch-all
@@ -44,13 +44,13 @@ Provides-Extra: pytorch-dev
 Provides-Extra: app-extra
 Provides-Extra: app-all
 Provides-Extra: app-dev
+Provides-Extra: test
 Provides-Extra: strategies
-Provides-Extra: examples
 Provides-Extra: extra
-Provides-Extra: test
+Provides-Extra: examples
+Provides-Extra: cloud
 Provides-Extra: ui
 Provides-Extra: components
-Provides-Extra: cloud
 Provides-Extra: all
 Provides-Extra: dev
 License-File: LICENSE
@@ -73,7 +73,7 @@ ______________________________________________________________________
   <a href="https://lightning.ai/docs/pytorch/stable/">PyTorch Lightning</a> •
   <a href="https://lightning.ai/docs/fabric/stable/">Fabric</a> •
   <a href="https://lightning.ai/docs/app/stable/">Lightning Apps</a> •
-  <a href="https://pytorch-lightning.readthedocs.io/en/2.0.3">Docs</a> •
+  <a href="https://pytorch-lightning.readthedocs.io/en/2.0.4">Docs</a> •
   <a href="#community">Community</a> •
   <a href="https://lightning.ai/docs/pytorch/stable/generated/CONTRIBUTING.html">Contribute</a> •
 </p>
@@ -85,7 +85,7 @@ ______________________________________________________________________
 [![PyPI Status](https://pepy.tech/badge/pytorch-lightning)](https://pepy.tech/project/pytorch-lightning)
 [![Conda](https://img.shields.io/conda/v/conda-forge/lightning?label=conda&color=success)](https://anaconda.org/conda-forge/lightning)
 [![DockerHub](https://img.shields.io/docker/pulls/pytorchlightning/pytorch_lightning.svg)](https://hub.docker.com/r/pytorchlightning/pytorch_lightning)
-[![codecov](https://codecov.io/gh/Lightning-AI/lightning/release/2.0.3/graph/badge.svg?token=SmzX8mnKlA)](https://codecov.io/gh/Lightning-AI/lightning)
+[![codecov](https://codecov.io/gh/Lightning-AI/lightning/release/2.0.4/graph/badge.svg?token=SmzX8mnKlA)](https://codecov.io/gh/Lightning-AI/lightning)
 [![Discord](https://img.shields.io/discord/1077906959069626439?style=plastic)](https://discord.gg/VptPCZkGNa)
 ![GitHub commit activity](https://img.shields.io/github/commit-activity/w/lightning-ai/lightning)
@@ -488,10 +488,10 @@ Lightning is rigorously tested across multiple CPUs, GPUs, TPUs, IPUs, and HPUs
 |       System / PyTorch ver.        |                                                                                              1.11                                                                                               |                                                                                                              1.12                                                                                                               | 1.13                                                                                                                                                                                                                            | 2.0  |
 | :--------------------------------: | :---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | ---- |
-|        Linux py3.9 \[GPUs\]        |                                                                                                -                                                                                                | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(GPUs)?branchName=refs%2Ftags%2F2.0.3>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(GPUs)?branchName=refs%2Ftags%2F2.0.3>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) | Soon |
+|        Linux py3.9 \[GPUs\]        |                                                                                                -                                                                                                | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(GPUs)?branchName=refs%2Ftags%2F2.0.4>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(GPUs)?branchName=refs%2Ftags%2F2.0.4>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=24&branchName=master) | Soon |
 |        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)                     |                                                                                                                                                                                                                                 | Soon |
-|        Linux py3.8 \[IPUs\]        |                                                                                                -                                                                                                |                                                                                                                -                                                                                                                | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(IPUs)?branchName=refs%2Ftags%2F2.0.3>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=25&branchName=master) | Soon |
-|        Linux py3.8 \[HPUs\]        |                                                                                                -                                                                                                |                                                                                                                -                                                                                                                | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(HPUs)?branchName=refs%2Ftags%2F2.0.3>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=26&branchName=master) | Soon |
+|        Linux py3.8 \[IPUs\]        |                                                                                                -                                                                                                |                                                                                                                -                                                                                                                | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(IPUs)?branchName=refs%2Ftags%2F2.0.4>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=25&branchName=master) | Soon |
+|        Linux py3.8 \[HPUs\]        |                                                                                                -                                                                                                |                                                                                                                -                                                                                                                | [![Build Status](<https://dev.azure.com/Lightning-AI/lightning/_apis/build/status/pytorch-lightning%20(HPUs)?branchName=refs%2Ftags%2F2.0.4>)](https://dev.azure.com/Lightning-AI/lightning/_build/latest?definitionId=26&branchName=master) | Soon |
 |  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)                                 | Soon |
 |   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)                                 | Soon |
 | Windows (multiple Python versions) | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml) |                 [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                 | [![Test PyTorch](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml/badge.svg)](https://github.com/Lightning-AI/lightning/actions/workflows/ci-tests-pytorch.yml)                                 | Soon |

{lightning-2.0.3 → lightning-2.0.4}/requirements/app/base.txt RENAMED Viewed

@@ -8,22 +8,22 @@ croniter >=1.3.0, <1.4.0  # strict; TODO: for now until we find something more r
 traitlets >=5.3.0, <5.10.0
 arrow >=1.2.0, <1.2.4
 lightning-utilities >=0.7.0, <0.9.0
-beautifulsoup4 >=4.8.0, <4.11.2
+beautifulsoup4 >=4.8.0, <4.12.3
 inquirer >=2.10.0, <=3.1.3
 psutil <5.9.5
 click <=8.1.3
 python-multipart>=0.0.5, <=0.0.6
-fastapi >=0.69.0, <0.89.0  # strict; TODO: broken serializations
+fastapi >=0.92.0, <0.98.0
 starlette  # https://fastapi.tiangolo.com/deployment/versions/#about-starlette
 pydantic >=1.7.4, <2.0.0  # https://fastapi.tiangolo.com/deployment/versions/#about-pydantic
 dateutils <=0.6.12
 Jinja2 <=3.1.2
 PyYAML <=6.0
-requests <2.28.3
-rich >=12.3.0, <=13.0.1
-urllib3 <=1.26.13
+requests <2.31.1
+rich >=12.3.0, <=13.4.2
+urllib3 <=2.0.2
 uvicorn <=0.22.0
-websocket-client <1.5.2
+websocket-client <1.5.3
 websockets <=10.4

lightning-2.0.4/requirements/app/cloud.txt ADDED Viewed

@@ -0,0 +1,4 @@
+redis >=4.0.1, <=4.5.5
+docker >=5.0.0, <6.1.3
+s3fs >=2022.5.0, <2023.6.1
+# setuptools==59.5.0

{lightning-2.0.3 → lightning-2.0.4}/requirements/app/test.txt RENAMED Viewed

@@ -3,10 +3,10 @@ pytest ==7.3.1
 pytest-timeout ==2.1.0
 pytest-cov ==4.0.0
 pytest-doctestplus >=0.9.0
-pytest-asyncio ==0.20.3
+pytest-asyncio ==0.21.0
 pytest-rerunfailures <=11.1.2
-playwright ==1.32.1
-httpx
+playwright ==1.35.0
+httpx ==0.24.1
 trio <0.22.0  # strict https://github.com/python-trio/trio/pull/2213
 pympler
 psutil

{lightning-2.0.3 → lightning-2.0.4}/requirements/app/ui.txt RENAMED Viewed

@@ -1,2 +1,2 @@
 streamlit >=1.13.0, <1.22.1
-panel >=0.12.7, <=0.13.1
+panel >=1.0.0, <=1.0.2

{lightning-2.0.3 → lightning-2.0.4}/requirements/base.txt RENAMED Viewed

@@ -7,7 +7,7 @@ click<9.0
 croniter<1.4.0,>=1.3.0  # strict
 dateutils<1.0
 deepdiff<7.0,>=5.7.0
-fastapi<0.89.0,>=0.69.0  # strict
+fastapi<1.0,>=0.92.0
 fsspec<2023.0,>=2022.5.0
 fsspec[http]<2024.0,>2021.06.0
 inquirer<4.0,>=2.10.0
@@ -28,7 +28,7 @@ torchmetrics<1.0,>=0.7.0
 tqdm<5.0,>=4.57.0
 traitlets<6.0,>=5.3.0
 typing-extensions<5.0,>=4.0.0
-urllib3<2.0
+urllib3<3.0
 uvicorn<1.0
 websocket-client<2.0
 websockets<11.0

{lightning-2.0.3 → lightning-2.0.4}/requirements/fabric/base.txt RENAMED Viewed

@@ -2,7 +2,7 @@
 #  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.24.4
-torch >=1.11.0, <=2.0.0
+torch >=1.11.0, <=2.0.1
 fsspec[http]>2021.06.0, <2023.5.0
 packaging >=17.1, <=23.0
 typing-extensions >=4.0.0, <=4.4.0

{lightning-2.0.3 → lightning-2.0.4}/requirements/fabric/test.txt RENAMED Viewed

@@ -3,4 +3,4 @@ pytest ==7.3.1
 pytest-cov ==4.0.0
 pytest-rerunfailures ==10.3
 click ==8.1.3
-tensorboardX >=2.2, <=2.5.1  # min version is set by torch.onnx missing attribute
+tensorboardX >=2.2, <=2.6.1  # min version is set by torch.onnx missing attribute

{lightning-2.0.3 → lightning-2.0.4}/requirements/pytorch/base.txt RENAMED Viewed

@@ -2,7 +2,7 @@
 #  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.24.4
-torch >=1.11.0, <=2.0.0
+torch >=1.11.0, <=2.0.1
 tqdm >=4.57.0, <4.66.0
 PyYAML >=5.4, <=6.0
 fsspec[http] >2021.06.0, <2023.5.0

{lightning-2.0.3 → lightning-2.0.4}/requirements/pytorch/examples.txt RENAMED Viewed

@@ -2,6 +2,6 @@
 #  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
 torchvision >=0.12.0, <=0.15.2
 gym[classic_control] >=0.17.0, <0.26.3
-ipython[all] <8.7.1
+ipython[all] <8.14.1
 torchmetrics >=0.10.0, <0.12.0
 lightning-utilities >=0.7.0, <0.9.0

{lightning-2.0.3 → lightning-2.0.4}/requirements/pytorch/extra.txt RENAMED Viewed

@@ -6,5 +6,5 @@ matplotlib>3.1, <3.6.2
 omegaconf >=2.0.5, <2.4.0
 hydra-core >=1.0.5, <1.4.0
 jsonargparse[signatures] >=4.18.0, <4.22.0
-rich >=12.3.0, <=13.0.1
-tensorboardX >=2.2, <=2.5.1  # min version is set by torch.onnx missing attribute
+rich >=12.3.0, <=13.4.2
+tensorboardX >=2.2, <=2.6.1  # min version is set by torch.onnx missing attribute

{lightning-2.0.3 → lightning-2.0.4}/requirements/pytorch/test.txt RENAMED Viewed

@@ -8,10 +8,10 @@ pytest-rerunfailures ==10.3
 cloudpickle >=1.3, <2.3.0
 scikit-learn >0.22.1, <1.2.1
 onnx <1.15.0
-onnxruntime <1.14.0
+onnxruntime <1.16.0
 psutil <5.9.5 # for `DeviceStatsMonitor`
-pandas >1.0, <2.0.2  # needed in benchmarks
-fastapi <0.87.0  # for `ServableModuleValidator`
+pandas >1.0, <2.0.3  # needed in benchmarks
+fastapi <0.98.0  # for `ServableModuleValidator`
 uvicorn <0.22.1  # for `ServableModuleValidator`
 tensorboard >=2.9.1, <2.14.0  # for `TensorBoardLogger`

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/app/CHANGELOG.md RENAMED Viewed

@@ -5,6 +5,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
+## [2.0.4] - 2023-06-22
+### Fixed
+- bumped several dependencies to address security volnebilities.
 ## [2.0.3] - 2023-06-07
 - Added the property `LightningWork.public_ip` that exposes the public IP of the `LightningWork` instance ([#17742](https://github.com/Lightning-AI/lightning/pull/17742))

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/app/cli/react-ui-template/ui/package.json RENAMED Viewed

@@ -24,7 +24,7 @@
     "@vitejs/plugin-react": "^1.0.7",
     "prettier": "^2.5.1",
     "typescript": "^4.5.4",
-    "vite": "^2.8.6"
+    "vite": "^2.9.16"
   },
   "main": "index.js",
   "license": "MIT"

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/app/cli/react-ui-template/ui/yarn.lock RENAMED Viewed

@@ -1050,10 +1050,10 @@ resolve@^1.12.0, resolve@^1.22.0:
     path-parse "^1.0.7"
     supports-preserve-symlinks-flag "^1.0.0"
-rollup@^2.59.0:
-  version "2.75.7"
-  resolved "https://registry.yarnpkg.com/rollup/-/rollup-2.75.7.tgz#221ff11887ae271e37dcc649ba32ce1590aaa0b9"
-  integrity sha512-VSE1iy0eaAYNCxEXaleThdFXqZJ42qDBatAwrfnPlENEZ8erQ+0LYX4JXOLPceWfZpV1VtZwZ3dFCuOZiSyFtQ==
+"rollup@>=2.59.0 <2.78.0":
+  version "2.77.3"
+  resolved "https://registry.yarnpkg.com/rollup/-/rollup-2.77.3.tgz#8f00418d3a2740036e15deb653bed1a90ee0cc12"
+  integrity sha512-/qxNTG7FbmefJWoeeYJFbHehJ2HNWnjkAFRKzWN/45eNBBF/r8lo992CwcJXEzyVxs5FmfId+vTSTQDb+bxA+g==
   optionalDependencies:
     fsevents "~2.3.2"
@@ -1120,15 +1120,15 @@ update-browserslist-db@^1.0.4:
     escalade "^3.1.1"
     picocolors "^1.0.0"
-vite@^2.8.6:
-  version "2.9.13"
-  resolved "https://registry.yarnpkg.com/vite/-/vite-2.9.13.tgz#859cb5d4c316c0d8c6ec9866045c0f7858ca6abc"
-  integrity sha512-AsOBAaT0AD7Mhe8DuK+/kE4aWYFMx/i0ZNi98hJclxb4e0OhQcZYUrvLjIaQ8e59Ui7txcvKMiJC1yftqpQoDw==
+vite@^2.9.16:
+  version "2.9.16"
+  resolved "https://registry.yarnpkg.com/vite/-/vite-2.9.16.tgz#daf7ba50f5cc37a7bf51b118ba06bc36e97898e9"
+  integrity sha512-X+6q8KPyeuBvTQV8AVSnKDvXoBMnTx8zxh54sOwmmuOdxkjMmEJXH2UEchA+vTMps1xw9vL64uwJOWryULg7nA==
   dependencies:
     esbuild "^0.14.27"
     postcss "^8.4.13"
     resolve "^1.22.0"
-    rollup "^2.59.0"
+    rollup ">=2.59.0 <2.78.0"
   optionalDependencies:
     fsevents "~2.3.2"

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/app/core/api.py RENAMED Viewed

@@ -231,7 +231,7 @@ def _get_component_by_name(component_name: str, state: dict) -> Union[LightningF
 @fastapi_service.get("/api/v1/layout", response_class=JSONResponse)
-async def get_layout() -> Mapping:
+async def get_layout() -> str:
     with lock:
         x_lightning_session_uuid = TEST_SESSION_UUID
         state = global_app_state_store.get_app_state(x_lightning_session_uuid)
@@ -240,7 +240,7 @@ async def get_layout() -> Mapping:
         for la in layout:
             if la["content"].startswith("root."):
                 la["content"] = _get_component_by_name(la["content"], state)
-        return layout
+        return json.dumps(layout)
 @fastapi_service.get("/api/v1/spec", response_class=JSONResponse)

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/fabric/CHANGELOG.md RENAMED Viewed

@@ -5,6 +5,14 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
+## [2.0.4] - 2023-06-22
+### Fixed
+- Fixed validation of parameters of `plugins.precision.MixedPrecision` ([#17687](https://github.com/Lightning-AI/lightning/pull/17687))
+- Fixed an issue with hpu imports leading to performance degradation  ([#17788](https://github.com/Lightning-AI/lightning/pull/17788))
 ## [2.0.3] - 2023-06-07
 - Added support for `Callback` registration through entry points ([#17756](https://github.com/Lightning-AI/lightning/pull/17756))

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/fabric/plugins/precision/amp.py RENAMED Viewed

@@ -41,6 +41,12 @@ class MixedPrecision(Precision):
         device: str,
         scaler: Optional[torch.cuda.amp.GradScaler] = None,
     ) -> None:
+        if precision not in ("16-mixed", "bf16-mixed"):
+            raise ValueError(
+                f"Passed `{type(self).__name__}(precision={precision!r})`."
+                " Precision must be '16-mixed' or 'bf16-mixed'."
+            )
         self.precision = cast(Literal["16-mixed", "bf16-mixed"], str(precision))
         if scaler is None and self.precision == "16-mixed":
             with _patch_cuda_is_available():

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/fabric/utilities/distributed.py RENAMED Viewed

@@ -6,7 +6,7 @@ from typing import Any, Iterable, Iterator, List, Optional, Sized, Tuple, Union
 import torch
 import torch.nn.functional as F
-from lightning_utilities.core.imports import module_available
+from lightning_utilities.core.imports import package_available
 from torch import Tensor
 from torch.utils.data import Dataset, DistributedSampler, Sampler
@@ -138,16 +138,17 @@ def _sync_ddp(result: Tensor, group: Optional[Any] = None, reduce_op: Optional[U
         op = reduce_op
     # WA for HPU. HPU doesn't support Long types, forcefully set it to float
-    if module_available("habana_frameworks.torch.utils.library_loader"):
-        from habana_frameworks.torch.utils.library_loader import is_habana_avaialble
-        if (
-            is_habana_avaialble()
-            and os.environ.get("HCCL_DISTRIBUTED_BACKEND") == "1"
-            and result.type() in ("torch.LongTensor", "torch.hpu.LongTensor")
-        ):
-            rank_zero_info("Long tensor unsupported on HPU, casting to float")
-            result = result.float()
+    if (
+        package_available("habana_frameworks")
+        and os.environ.get("HCCL_DISTRIBUTED_BACKEND") == "1"
+        and result.type()
+        in (
+            "torch.LongTensor",
+            "torch.hpu.LongTensor",
+        )
+    ):
+        rank_zero_info("Long tensor unsupported on HPU, casting to float")
+        result = result.float()
     # Sync all processes before reduction
     torch.distributed.barrier(group=group)

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/pytorch/CHANGELOG.md RENAMED Viewed

@@ -4,6 +4,20 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
+## [2.0.4] - 2023-06-22
+### Changed
+- Changes to the `NeptuneLogger` ([#16761](https://github.com/Lightning-AI/lightning/pull/16761)):
+  * It now supports neptune-client 0.16.16 and neptune >=1.0, and we have replaced the `log()` method with `append()` and `extend()`.
+  * It now accepts a namespace `Handler` as an alternative to `Run` for the `run` argument. This means that you can call it like `NeptuneLogger(run=run["some/namespace"])` to log everything to the `some/namespace/` location of the run.
+### Fixed
+- Fixed validation of parameters of `plugins.precision.MixedPrecisionPlugin` ([#17687](https://github.com/Lightning-AI/lightning/pull/17687))
+- Fixed deriving default map location in `LightningModule.load_from_checkpoint` when there is extra state ([#17812](https://github.com/Lightning-AI/lightning/pull/17812))
 ## [2.0.3] - 2023-06-07

{lightning-2.0.3 → lightning-2.0.4}/src/lightning/pytorch/core/saving.py RENAMED Viewed

@@ -25,6 +25,7 @@ from pathlib import Path
 from typing import Any, Callable, Dict, IO, Optional, Type, Union
 from warnings import warn
+import torch
 import yaml
 from lightning_utilities.core.apply_func import apply_to_collection
@@ -86,13 +87,14 @@ def _load_from_checkpoint(
     if issubclass(cls, pl.LightningDataModule):
         return _load_state(cls, checkpoint, **kwargs)
     if issubclass(cls, pl.LightningModule):
-        storage = _load_state(cls, checkpoint, strict=strict, **kwargs)
+        model = _load_state(cls, checkpoint, strict=strict, **kwargs)
         state_dict = checkpoint["state_dict"]
         if not state_dict:
             raise ValueError(f"The state dict in {checkpoint_path!r} contains no parameters.")
-        map_location = list(state_dict.values())[0].device
-        assert isinstance(storage, pl.LightningModule)
-        return storage.to(map_location)
+        device = next((t for t in state_dict.values() if isinstance(t, torch.Tensor)), torch.tensor(0)).device
+        assert isinstance(model, pl.LightningModule)
+        return model.to(device)
     raise NotImplementedError(f"Unsupported {cls}")

lightning 2.0.3__tar.gz → 2.0.4__tar.gz

lightning 2.0.3tar.gz → 2.0.4tar.gz