Skip to content

Commit

Permalink
[FEATURE] Print model ID for trainer run
Browse files Browse the repository at this point in the history
GitOrigin-RevId: 0d3a057dd9e329576c9898f63b171b019a9409a5
  • Loading branch information
mckornfield committed Sep 27, 2024
1 parent c6986dd commit 50c06c6
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/gretel_trainer/benchmark/gretel/strategy_sdk.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import copy
import gzip
import logging

from pathlib import Path
from typing import Optional
Expand All @@ -20,6 +21,8 @@
from gretel_trainer.benchmark.job_spec import JobSpec
from gretel_trainer.benchmark.sdk_extras import await_job

logger = logging.getLogger(__name__)


class GretelSDKStrategy:
def __init__(
Expand Down Expand Up @@ -67,6 +70,7 @@ def train(self) -> None:
_model = self.project.create_model_obj(
model_config=model_config, data_source=data_source
)
logging.info("Created model(name=%s,id=%s)", _model.name, _model.id)
# Calling this in lieu of submit_cloud() is supposed to avoid
# artifact upload. Doesn't work for more recent client versions!
self.model = _model.submit(runner_mode=RunnerMode.CLOUD)
Expand Down

0 comments on commit 50c06c6

Please sign in to comment.