Make apply_to_collection importable from top level (#120)

Lightning-AI · Mar 8, 2023 · ca500fd · ca500fd
1 parent 1be780f
commit ca500fd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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",

diff --git 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",