Skip to content

Commit dadd372

Browse files
authored
Enable EULA acceptance in model customization tests
Set accept_eula to True in model builder to fix tests
1 parent 6f73fbf commit dadd372

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

sagemaker-serve/tests/integ/test_model_customization_deployment.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ def test_build_from_training_job(self, training_job_name):
119119

120120
training_job = TrainingJob.get(training_job_name=training_job_name)
121121
model_builder = ModelBuilder(model=training_job)
122+
model_builder.accept_eula = True
122123
model = model_builder.build(model_name=f"test-model-{int(time.time())}-{random.randint(100, 10000)}")
123124

124125
assert model is not None
@@ -139,6 +140,7 @@ def test_deploy_from_training_job(self, training_job_name, endpoint_name, cleanu
139140

140141
training_job = TrainingJob.get(training_job_name=training_job_name)
141142
model_builder = ModelBuilder(model=training_job, instance_type="ml.g5.4xlarge")
143+
model_builder.accept_eula = True
142144
model_builder.build(model_name=f"test-model-{int(time.time())}-{random.randint(100, 10000)}")
143145

144146
peft_type = model_builder._fetch_peft()
@@ -187,6 +189,7 @@ def test_build_from_model_package(self, model_package_arn):
187189

188190
model_package = ModelPackage.get(model_package_name=model_package_arn)
189191
model_builder = ModelBuilder(model=model_package)
192+
model_builder.accept_eula = True
190193
model = model_builder.build()
191194

192195
assert model is not None
@@ -201,6 +204,7 @@ def test_deploy_from_model_package(self, model_package_arn, cleanup_endpoints):
201204
model_package = ModelPackage.get(model_package_name=model_package_arn)
202205
endpoint_name = f"e2e-{int(time.time())}-{random.randint(100, 10000)}"
203206
model_builder = ModelBuilder(model=model_package)
207+
model_builder.accept_eula = True
204208
model_builder.build()
205209
endpoint = model_builder.deploy(endpoint_name=endpoint_name)
206210

@@ -220,6 +224,7 @@ def test_instance_type_from_recipe(self, training_job_name):
220224

221225
training_job = TrainingJob.get(training_job_name=training_job_name)
222226
model_builder = ModelBuilder(model=training_job)
227+
model_builder.accept_eula = True
223228
model_builder.build()
224229

225230
assert model_builder.instance_type is not None

0 commit comments

Comments
 (0)