@@ -35,15 +35,16 @@ def test_training(sagemaker_session, ecr_image, instance_type, framework_version
3535
3636 sm_client = sagemaker_session .sagemaker_client
3737
38- experiment_name = f "tf-container-integ-test-{ int (time .time ())} "
38+ experiment_name = "tf-container-integ-test-{}" . format ( int (time .time ()))
3939
4040 experiment = Experiment .create (
4141 experiment_name = experiment_name ,
4242 description = "Integration test experiment from sagemaker-tf-container" ,
4343 sagemaker_boto_client = sm_client ,
4444 )
4545
46- trial_name = f"tf-container-integ-test-{ int (time .time ())} "
46+ trial_name = "tf-container-integ-test-{}" .format (int (time .time ()))
47+
4748 trial = Trial .create (
4849 experiment_name = experiment_name , trial_name = trial_name , sagemaker_boto_client = sm_client
4950 )
@@ -80,16 +81,12 @@ def test_training(sagemaker_session, ecr_image, instance_type, framework_version
8081 trial_component_summary = trial_components [0 ]
8182 trial_component = TrialComponent .load (
8283 trial_component_name = trial_component_summary .trial_component_name ,
83- sagemaker_boto_name = sm_client ,
84+ sagemaker_boto_client = sm_client ,
8485 )
8586
8687 # associate the trial component with the trial
8788 trial .add_trial_component (trial_component )
8889
89- # verify association
90- associated_trial_components = list (trial .list_trial_components ())
91- assert len (associated_trial_components ) == 1
92-
9390 # cleanup
9491 trial .remove_trial_component (trial_component_summary .trial_component_name )
9592 trial_component .delete ()
0 commit comments