Skip to content

Commit

Permalink
Add flavor as self reference
Browse files Browse the repository at this point in the history
  • Loading branch information
mn-mikke committed Nov 13, 2023
1 parent aad61ae commit 77c4391
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion h2o-py-mlflow-flavor/h2o_mlflow_flavor/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@
_validate_and_prepare_target_save_path,
)
from mlflow.tracking.artifact_utils import _download_artifact_from_uri
from mlflow.utils.requirements_utils import _get_pinned_requirement

_logger = logging.getLogger(__name__)

Expand All @@ -47,7 +48,7 @@ def get_default_pip_requirements():
Calls to :func:`save_model()` and :func:`log_model()` produce a pip environment
that, at minimum, contains these requirements.
"""
return []
return [_get_pinned_requirement("h2o_mlflow_flavor")]


def get_default_conda_env():
Expand Down

0 comments on commit 77c4391

Please sign in to comment.