No public description

PiperOrigin-RevId: 581987803
google · Nov 17, 2023 · 412801d · 412801d
1 parent c41d4b4
commit 412801d
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/flax/__init__.py b/flax/__init__.py
@@ -14,18 +14,26 @@
 
 """Flax API."""
 
+# pylint: disable=g-import-not-at-top
 # pyformat: disable
 
-from .configurations import (
-    config as config,
-)
+from flax import configurations
+config: configurations.Config = configurations.config
+del configurations
 
-from . import core
-from . import jax_utils
-from . import linen
-from . import serialization
-from . import traverse_util
+from flax import core
+from flax import jax_utils
+from flax import linen
+from flax import serialization
+from flax import traverse_util
+
+from flax import version
+__version__: str = version.__version__
+del version
 
 # DO NOT REMOVE - Marker for internal deprecated API.
+
 # DO NOT REMOVE - Marker for internal logging.
-from .version import __version__
+
+# pyformat: enable
+# pylint: enable=g-import-not-at-top