Skip to content

Commit a7d0d55

Browse files
authored
Unified config imports to match docs (#3743)
* Fixed imports instead of docs * Radically cut back on imports
1 parent 96f0ee3 commit a7d0d55

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

src/zenml/config/__init__.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,23 @@
2323
deserialization of the configuration options that are stored in the file in
2424
order to persist the configuration across sessions.
2525
"""
26-
from zenml.config.docker_settings import DockerSettings
27-
from zenml.config.resource_settings import ResourceSettings
26+
from zenml.config.docker_settings import (
27+
DockerSettings,
28+
PythonPackageInstaller,
29+
PythonEnvironmentExportMethod,
30+
)
31+
from zenml.config.resource_settings import ResourceSettings, ByteUnit
2832
from zenml.config.retry_config import StepRetryConfig
33+
from zenml.config.schedule import Schedule
34+
from zenml.config.store_config import StoreConfiguration
2935

3036
__all__ = [
3137
"DockerSettings",
38+
"PythonPackageInstaller",
39+
"PythonEnvironmentExportMethod",
3240
"ResourceSettings",
41+
"ByteUnit",
3342
"StepRetryConfig",
43+
"Schedule",
44+
"StoreConfiguration",
3445
]

0 commit comments

Comments
 (0)