Skip to content

Commit 8656d6a

Browse files
author
Syed Jafri
committed
debug: remove unneeded fields for rvlr_nova test + logging in finetuning utils
1 parent adef22f commit 8656d6a

2 files changed

Lines changed: 11 additions & 5 deletions

File tree

sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,12 +105,22 @@ def _resolve_mlflow_resource_arn(sagemaker_session, mlflow_resource_arn: Optiona
105105
return mlflow_resource_arn
106106

107107
try:
108+
target_region = sagemaker_session.boto_session.region_name
109+
logger.info("Resolving MLflow app for region: %s", target_region)
110+
108111
mlflow_apps = MlflowApp.get_all(
109112
session=sagemaker_session.boto_session,
110-
region=sagemaker_session.boto_session.region_name
113+
region=target_region
111114
)
112115

113116
mlflow_apps_list = list(mlflow_apps)
117+
logger.info("Found %d MLflow apps from ListMlflowApps API (region=%s):", len(mlflow_apps_list), target_region)
118+
for app in mlflow_apps_list:
119+
logger.info(" App ARN: %s | status: %s | account_default_status: %s",
120+
getattr(app, 'arn', 'N/A'),
121+
getattr(app, 'status', 'N/A'),
122+
getattr(app, 'account_default_status', 'N/A'))
123+
114124
current_domain_id = _get_current_domain_id(sagemaker_session)
115125

116126
# Check for domain match

sagemaker-train/tests/integ/train/test_rlvr_trainer_integration.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,10 +111,6 @@ def test_rlvr_trainer_nova_workflow(sagemaker_session_us_east_1):
111111
accept_eula=True,
112112
sagemaker_session=sagemaker_session_us_east_1
113113
)
114-
rlvr_trainer.hyperparameters.data_s3_path = 's3://example-bucket'
115-
116-
rlvr_trainer.hyperparameters.reward_lambda_arn = 'arn:aws:lambda:us-east-1:729646638167:function:rlvr-nova-reward-function'
117-
118114
training_job = rlvr_trainer.train(wait=False)
119115

120116
# Manual wait loop

0 commit comments

Comments
 (0)