Skip to content

Commit 2fb0ac1

Browse files
author
Roja Reddy Sareddy
committed
Fix: Add allow list for bedrock eval models
1 parent c43a863 commit 2fb0ac1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sagemaker-train/src/sagemaker/train/evaluate/llm_as_judge_evaluator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from .base_evaluator import BaseEvaluator
1414
from sagemaker.core.telemetry.telemetry_logging import _telemetry_emitter
1515
from sagemaker.core.telemetry.constants import Feature
16+
from sagemaker.train.constants import _ALLOWED_EVALUATOR_MODELS
1617

1718
_logger = logging.getLogger(__name__)
1819

@@ -148,7 +149,6 @@ def _validate_model_compatibility(cls, v, values):
148149
@validator('evaluator_model')
149150
def _validate_evaluator_model(cls, v, values):
150151
"""Validate evaluator_model is allowed and check region compatibility."""
151-
from sagemaker.train.constants import _ALLOWED_EVALUATOR_MODELS
152152

153153
if v not in _ALLOWED_EVALUATOR_MODELS:
154154
raise ValueError(

0 commit comments

Comments
 (0)