Skip to content

Commit 4147d4a

Browse files
committed
fix: comment out non-existent mlflow_resource_arn in base_model_only benchmark test
1 parent 076b890 commit 4147d4a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎sagemaker-train/tests/integ/train/test_benchmark_evaluator.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -307,7 +307,7 @@ def test_benchmark_evaluation_base_model_only(self):
307307
benchmark=Benchmark.MMLU,
308308
model=BASE_MODEL_ONLY_CONFIG["base_model_id"],
309309
s3_output_path=BASE_MODEL_ONLY_CONFIG["s3_output_path"],
310-
mlflow_resource_arn=BASE_MODEL_ONLY_CONFIG["mlflow_tracking_server_arn"],
310+
# mlflow_resource_arn=BASE_MODEL_ONLY_CONFIG["mlflow_tracking_server_arn"],
311311
base_eval_name="integ-test-base-model-only",
312312
# Note: model_package_group not needed for JumpStart models
313313
)

0 commit comments

Comments
 (0)