@@ -35,15 +35,16 @@ def test_training(sagemaker_session, ecr_image, instance_type, framework_version
35
35
36
36
sm_client = sagemaker_session .sagemaker_client
37
37
38
- experiment_name = f "tf-container-integ-test-{ int (time .time ())} "
38
+ experiment_name = "tf-container-integ-test-{}" . format ( int (time .time ()))
39
39
40
40
experiment = Experiment .create (
41
41
experiment_name = experiment_name ,
42
42
description = "Integration test experiment from sagemaker-tf-container" ,
43
43
sagemaker_boto_client = sm_client ,
44
44
)
45
45
46
- trial_name = f"tf-container-integ-test-{ int (time .time ())} "
46
+ trial_name = "tf-container-integ-test-{}" .format (int (time .time ()))
47
+
47
48
trial = Trial .create (
48
49
experiment_name = experiment_name , trial_name = trial_name , sagemaker_boto_client = sm_client
49
50
)
@@ -80,16 +81,12 @@ def test_training(sagemaker_session, ecr_image, instance_type, framework_version
80
81
trial_component_summary = trial_components [0 ]
81
82
trial_component = TrialComponent .load (
82
83
trial_component_name = trial_component_summary .trial_component_name ,
83
- sagemaker_boto_name = sm_client ,
84
+ sagemaker_boto_client = sm_client ,
84
85
)
85
86
86
87
# associate the trial component with the trial
87
88
trial .add_trial_component (trial_component )
88
89
89
- # verify association
90
- associated_trial_components = list (trial .list_trial_components ())
91
- assert len (associated_trial_components ) == 1
92
-
93
90
# cleanup
94
91
trial .remove_trial_component (trial_component_summary .trial_component_name )
95
92
trial_component .delete ()
0 commit comments