@@ -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