pytorch-lightning 2.2.2__tar.gz → 2.2.4__tar.gz

{pytorch-lightning-2.2.2/src/pytorch_lightning.egg-info → pytorch-lightning-2.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytorch-lightning
-Version: 2.2.2
+Version: 2.2.4
 Summary: PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
 Home-page: https://github.com/Lightning-AI/lightning
 Author: Lightning AI et al.
@@ -29,9 +29,9 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Provides-Extra: strategies
-Provides-Extra: examples
 Provides-Extra: test
 Provides-Extra: extra
+Provides-Extra: examples
 Provides-Extra: deepspeed
 Provides-Extra: all
 Provides-Extra: dev
@@ -66,7 +66,7 @@ ______________________________________________________________________
 [![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/pytorch-lightning/graph/badge.svg?token=SmzX8mnKlA)](https://codecov.io/gh/Lightning-AI/pytorch-lightning)
 
-[![ReadTheDocs](https://readthedocs.org/projects/pytorch-lightning/badge/?version=2.2.2)](https://lightning.ai/docs/pytorch/stable/)[![Discord](https://img.shields.io/discord/1077906959069626439?style=plastic)](https://discord.gg/VptPCZkGNa)
+[![ReadTheDocs](https://readthedocs.org/projects/pytorch-lightning/badge/?version=2.2.4)](https://lightning.ai/docs/pytorch/stable/)[![Discord](https://img.shields.io/discord/1077906959069626439?style=plastic)](https://discord.gg/VptPCZkGNa)
 [![license](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/Lightning-AI/lightning/blob/master/LICENSE)
 
 <!--

{pytorch-lightning-2.2.2 → pytorch-lightning-2.2.4}/src/lightning_fabric/utilities/logger.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
+import json
 from argparse import Namespace
 from dataclasses import asdict, is_dataclass
 from typing import Any, Dict, Mapping, MutableMapping, Optional, Union
@@ -132,6 +132,23 @@ def _sanitize_params(params: Dict[str, Any]) -> Dict[str, Any]:
     return params
 
 
+def _convert_json_serializable(params: Dict[str, Any]) -> Dict[str, Any]:
+    """Convert non-serializable objects in params to string."""
+    return {k: str(v) if not _is_json_serializable(v) else v for k, v in params.items()}
+
+
+def _is_json_serializable(value: Any) -> bool:
+    """Test whether a variable can be encoded as json."""
+    if value is None or isinstance(value, (bool, int, float, str, list, dict)):  # fast path
+        return True
+    try:
+        json.dumps(value)
+        return True
+    except (TypeError, OverflowError):
+        # OverflowError is raised if number is too large to encode
+        return False
+
+
 def _add_prefix(
     metrics: Mapping[str, Union[Tensor, float]], prefix: str, separator: str
 ) -> Mapping[str, Union[Tensor, float]]:

pytorch-lightning-2.2.4/src/lightning_fabric/version.info

@@ -0,0 +1 @@
+2.2.4

{pytorch-lightning-2.2.2 → pytorch-lightning-2.2.4}/src/pytorch_lightning/CHANGELOG.md

@@ -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.2.3] - 2024-04-23
+
+### Fixed
+
+- Fixed `WandbLogger.log_hyperparameters()` raising an error if hyperparameters are not JSON serializable ([#19769](https://github.com/Lightning-AI/pytorch-lightning/pull/19769))
+
+
 ## [2.2.2] - 2024-04-11
 
 ### Fixed

{pytorch-lightning-2.2.2 → pytorch-lightning-2.2.4}/src/pytorch_lightning/loggers/wandb.py

@@ -26,7 +26,12 @@ from lightning_utilities.core.imports import RequirementCache
 from torch import Tensor
 from typing_extensions import override
 
-from lightning_fabric.utilities.logger import _add_prefix, _convert_params, _sanitize_callable_params
+from lightning_fabric.utilities.logger import (
+    _add_prefix,
+    _convert_json_serializable,
+    _convert_params,
+    _sanitize_callable_params,
+)
 from lightning_fabric.utilities.types import _PATH
 from pytorch_lightning.callbacks.model_checkpoint import ModelCheckpoint
 from pytorch_lightning.loggers.logger import Logger, rank_zero_experiment
@@ -419,6 +424,7 @@ class WandbLogger(Logger):
     def log_hyperparams(self, params: Union[Dict[str, Any], Namespace]) -> None:  # type: ignore[override]
         params = _convert_params(params)
         params = _sanitize_callable_params(params)
+        params = _convert_json_serializable(params)
         self.experiment.config.update(params, allow_val_change=True)
 
     @override

pytorch-lightning-2.2.4/src/pytorch_lightning/version.info

@@ -0,0 +1 @@
+2.2.4

{pytorch-lightning-2.2.2 → pytorch-lightning-2.2.4/src/pytorch_lightning.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytorch-lightning
-Version: 2.2.2
+Version: 2.2.4
 Summary: PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
 Home-page: https://github.com/Lightning-AI/lightning
 Author: Lightning AI et al.
@@ -29,9 +29,9 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Provides-Extra: strategies
-Provides-Extra: examples
 Provides-Extra: test
 Provides-Extra: extra
+Provides-Extra: examples
 Provides-Extra: deepspeed
 Provides-Extra: all
 Provides-Extra: dev
@@ -66,7 +66,7 @@ ______________________________________________________________________
 [![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/pytorch-lightning/graph/badge.svg?token=SmzX8mnKlA)](https://codecov.io/gh/Lightning-AI/pytorch-lightning)
 
-[![ReadTheDocs](https://readthedocs.org/projects/pytorch-lightning/badge/?version=2.2.2)](https://lightning.ai/docs/pytorch/stable/)[![Discord](https://img.shields.io/discord/1077906959069626439?style=plastic)](https://discord.gg/VptPCZkGNa)
+[![ReadTheDocs](https://readthedocs.org/projects/pytorch-lightning/badge/?version=2.2.4)](https://lightning.ai/docs/pytorch/stable/)[![Discord](https://img.shields.io/discord/1077906959069626439?style=plastic)](https://discord.gg/VptPCZkGNa)
 [![license](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/Lightning-AI/lightning/blob/master/LICENSE)
 
 <!--

pytorch-lightning-2.2.4/src/version.info

@@ -0,0 +1 @@
+2.2.4

pytorch-lightning-2.2.2/src/lightning_fabric/version.info

@@ -1 +0,0 @@
-2.2.2

pytorch-lightning-2.2.2/src/pytorch_lightning/version.info

@@ -1 +0,0 @@
-2.2.2

pytorch-lightning-2.2.2/src/version.info

@@ -1 +0,0 @@
-2.2.2