From ab7e1788696eee356a444c409fbfac0189d02c91 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Wed, 8 Mar 2023 07:01:26 -0500 Subject: [PATCH 1/2] Make apply_to_collection importable from top level --- src/lightning_utilities/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/lightning_utilities/__init__.py b/src/lightning_utilities/__init__.py index 3b4ced87..83f1ea55 100644 --- a/src/lightning_utilities/__init__.py +++ b/src/lightning_utilities/__init__.py @@ -3,6 +3,7 @@ import os from lightning_utilities.__about__ import * # noqa: F401, F403 +from lightning_utilities.core.apply_func import apply_to_collection from lightning_utilities.core.enums import StrEnum from lightning_utilities.core.imports import compare_version, module_available from lightning_utilities.core.overrides import is_overridden @@ -13,6 +14,7 @@ __all__ = [ + "apply_to_collection", "StrEnum", "module_available", "compare_version", From 3d92238a1e71d807a6fd9e99140a2f7b35a03769 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Wed, 8 Mar 2023 13:04:55 +0100 Subject: [PATCH 2/2] core --- src/lightning_utilities/core/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/lightning_utilities/core/__init__.py b/src/lightning_utilities/core/__init__.py index be6f0c64..deda5d0e 100644 --- a/src/lightning_utilities/core/__init__.py +++ b/src/lightning_utilities/core/__init__.py @@ -1,10 +1,12 @@ """Core utilities.""" +from lightning_utilities.core.apply_func import apply_to_collection from lightning_utilities.core.enums import StrEnum from lightning_utilities.core.imports import compare_version, module_available from lightning_utilities.core.overrides import is_overridden from lightning_utilities.core.rank_zero import WarningCache __all__ = [ + "apply_to_collection", "StrEnum", "module_available", "compare_version",