Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Kubeflow dockerignore #249

Merged
merged 9 commits into from
Dec 21, 2021
Prev Previous commit
Next Next commit
Load components from correct repo
  • Loading branch information
schustmi committed Dec 20, 2021
commit dcfa8ac94282a2a84a6b7393db4e9f5a30911efe
4 changes: 3 additions & 1 deletion src/zenml/core/mapping_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@ def get_component_from_key(
Args:
key: Unique key.
mapping: Dict of type str -> UUIDSourceTuple.
repo_path: Path to the repo from which to load the component.

Returns:
An object which is a subclass of type BaseComponent.
Expand All @@ -66,12 +67,13 @@ def get_components_from_store(
Args:
store_name: Name of the store.
mapping: Dict of type str -> UUIDSourceTuple.
repo_path: Path to the repo from which to load the components.

Returns:
A dict of objects which are a subclass of type BaseComponent.
"""
store_dir = os.path.join(
zenml.io.utils.get_zenml_config_dir(),
zenml.io.utils.get_zenml_config_dir(repo_path),
store_name,
)
comps = {}
Expand Down