From 50c06c6c8e98153fd801908367b4dca91990f008 Mon Sep 17 00:00:00 2001 From: Matt Kornfield Date: Fri, 27 Sep 2024 11:33:51 -0700 Subject: [PATCH] [FEATURE] Print model ID for trainer run GitOrigin-RevId: 0d3a057dd9e329576c9898f63b171b019a9409a5 --- src/gretel_trainer/benchmark/gretel/strategy_sdk.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/gretel_trainer/benchmark/gretel/strategy_sdk.py b/src/gretel_trainer/benchmark/gretel/strategy_sdk.py index d8c07a2e..351e2402 100644 --- a/src/gretel_trainer/benchmark/gretel/strategy_sdk.py +++ b/src/gretel_trainer/benchmark/gretel/strategy_sdk.py @@ -1,5 +1,6 @@ import copy import gzip +import logging from pathlib import Path from typing import Optional @@ -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__( @@ -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)