Skip to content

Commit 12dde3e

Browse files
authored
Removing the deprecated log_xxx_metadata calls (#48)
* removing deprecated metadata calls * fixing the call
1 parent f67756a commit 12dde3e

File tree

4 files changed

+9
-8
lines changed

4 files changed

+9
-8
lines changed

.github/workflows/ci.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,5 +57,5 @@ jobs:
5757
with:
5858
stack-name: ${{ matrix.stack-name }}
5959
python-version: ${{ matrix.python-version }}
60-
ref-zenml: ${{ inputs.ref-zenml || 'develop' }}
60+
ref-zenml: ${{ inputs.ref-zenml || 'feature/followup-run-metadata' }}
6161
ref-template: ${{ inputs.ref-template || github.ref }}

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -125,13 +125,13 @@ To create parallel processing of computationally expensive operations we use a l
125125
<summary>Code snippet 💻</summary>
126126

127127
```python
128-
from zenml import log_artifact_metadata
128+
from zenml import log_metadata
129129
130130
score = accuracy_score(y_tst, y_pred)
131131
# log score along with output artifact as metadata
132-
log_artifact_metadata(
133-
output_name="hp_result",
134-
metric=float(score),
132+
log_metadata(
133+
metadata={"metric": float(score)},
134+
artifact_name="hp_result",
135135
)
136136
```
137137
</details>

template/steps/training/model_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import mlflow
66
import pandas as pd
77
from sklearn.base import ClassifierMixin
8-
from zenml import ArtifactConfig, log_artifact_metadata, step, get_step_context
8+
from zenml import ArtifactConfig, step, get_step_context
99
from zenml.client import Client
1010
from zenml.integrations.mlflow.experiment_trackers import MLFlowExperimentTracker
1111
from zenml.integrations.mlflow.steps.mlflow_registry import mlflow_register_model_step

template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from sklearn.metrics import accuracy_score
1010
from sklearn.model_selection import RandomizedSearchCV
1111
from utils import get_model_from_config
12-
from zenml import log_artifact_metadata, step
12+
from zenml import log_metadata, step
1313
from zenml.logger import get_logger
1414

1515
logger = get_logger(__name__)
@@ -79,9 +79,10 @@ def hp_tuning_single_search(
7979
y_pred = cv.predict(X_tst)
8080
score = accuracy_score(y_tst, y_pred)
8181
# log score along with output artifact as metadata
82-
log_artifact_metadata(
82+
log_metadata(
8383
metadata={"metric": float(score)},
8484
artifact_name="hp_result",
85+
infer_artifact=True,
8586
)
8687
### YOUR CODE ENDS HERE ###
8788
return cv.best_estimator_

0 commit comments

Comments
 (0)