Skip to content

Commit 8d87577

Browse files
committed
removing deprecated calls
1 parent dadf9be commit 8d87577

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.github/workflows/ci.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -56,5 +56,5 @@ jobs:
5656
with:
5757
stack-name: ${{ matrix.stack-name }}
5858
python-version: ${{ matrix.python-version }}
59-
ref-zenml: ${{ inputs.ref-zenml || 'develop' }}
59+
ref-zenml: ${{ inputs.ref-zenml || 'feature/followup-run-metadata' }}
6060
ref-template: ${{ inputs.ref-template || github.ref }}

template/steps/training/model_trainer.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
TrainingArguments,
1414
AutoModelForSequenceClassification,
1515
)
16-
from zenml import ArtifactConfig, log_artifact_metadata, step
16+
from zenml import ArtifactConfig, log_metadata, step
1717
from zenml.client import Client
1818
from zenml.integrations.mlflow.experiment_trackers import MLFlowExperimentTracker
1919
from zenml.logger import get_logger
@@ -135,7 +135,7 @@ def model_trainer(
135135
eval_results = trainer.evaluate(metric_key_prefix="")
136136

137137
# Log the evaluation results in model control plane
138-
log_artifact_metadata(
138+
log_metadata(
139139
metadata={"metrics": eval_results},
140140
artifact_name="model",
141141
)

0 commit comments

Comments
 (0)