@@ -19,8 +19,8 @@ def __init__(self, workspace_handler=None, **kwargs):
1919 def _log_create_experiment (self , response , success_msg_template , error_msg ):
2020 if response .ok :
2121 j = response .json ()
22- handle = j ["handle" ]
23- msg = success_msg_template .format (handle )
22+ id_ = j ["handle" ]
23+ msg = success_msg_template .format (id_ )
2424 self .logger .log (msg )
2525 else :
2626 try :
@@ -40,7 +40,7 @@ def execute(self, json_):
4040 response = self .api .post ("/experiments/" , json = json_ )
4141
4242 self ._log_create_experiment (response ,
43- "New experiment created with handle : {}" ,
43+ "New experiment created with ID : {}" ,
4444 "Unknown error while creating the experiment" )
4545
4646
@@ -52,18 +52,18 @@ def execute(self, json_):
5252
5353 response = self .api .post ("/experiments/create_and_start/" , json = json_ )
5454 self ._log_create_experiment (response ,
55- "New experiment created and started with handle : {}" ,
55+ "New experiment created and started with ID : {}" ,
5656 "Unknown error while creating/starting the experiment" )
5757
5858
59- def start_experiment (experiment_handle , api = experiments_api ):
60- url = "/experiments/{}/start/" .format (experiment_handle )
59+ def start_experiment (experiment_id , api = experiments_api ):
60+ url = "/experiments/{}/start/" .format (experiment_id )
6161 response = api .put (url )
6262 log_response (response , "Experiment started" , "Unknown error while starting the experiment" )
6363
6464
65- def stop_experiment (experiment_handle , api = experiments_api ):
66- url = "/experiments/{}/stop/" .format (experiment_handle )
65+ def stop_experiment (experiment_id , api = experiments_api ):
66+ url = "/experiments/{}/stop/" .format (experiment_id )
6767 response = api .put (url )
6868 log_response (response , "Experiment stopped" , "Unknown error while stopping the experiment" )
6969
@@ -73,9 +73,9 @@ def __init__(self, api=experiments_api, logger_=logger):
7373 self .api = api
7474 self .logger = logger_
7575
76- def execute (self , project_handles = None ):
77- project_handles = project_handles or []
78- params = self ._get_query_params (project_handles )
76+ def execute (self , project_ids = None ):
77+ project_ids = project_ids or []
78+ params = self ._get_query_params (project_ids )
7979 response = self .api .get ("/experiments/" , params = params )
8080
8181 try :
@@ -84,24 +84,24 @@ def execute(self, project_handles=None):
8484 self .logger .log_error_response (data )
8585 return
8686
87- experiments = self ._get_experiments_list (data , bool (project_handles ))
87+ experiments = self ._get_experiments_list (data , bool (project_ids ))
8888 except (ValueError , KeyError ) as e :
8989 self .logger .error ("Error while parsing response data: {}" .format (e ))
9090 else :
9191 self ._log_experiments_list (experiments )
9292
9393 @staticmethod
94- def _get_query_params (project_handles ):
94+ def _get_query_params (project_ids ):
9595 params = {"limit" : - 1 } # so the API sends back full list without pagination
96- for i , handle in enumerate (project_handles ):
96+ for i , experiment_id in enumerate (project_ids ):
9797 key = "projectHandle[{}]" .format (i )
98- params [key ] = handle
98+ params [key ] = experiment_id
9999
100100 return params
101101
102102 @staticmethod
103103 def _make_experiments_list_table (experiments ):
104- data = [("Name" , "Handle " , "Status" )]
104+ data = [("Name" , "ID " , "Status" )]
105105 for experiment in experiments :
106106 name = experiment ["templateHistory" ]["params" ].get ("name" )
107107 handle = experiment ["handle" ]
@@ -114,7 +114,7 @@ def _make_experiments_list_table(experiments):
114114
115115 @staticmethod
116116 def _get_experiments_list (data , filtered = False ):
117- if not filtered : # If filtering by projectHandle response data has different format...
117+ if not filtered : # If filtering by project ID response data has different format...
118118 return data ["data" ]
119119
120120 experiments = []
@@ -138,10 +138,10 @@ def _make_details_table(experiment):
138138 if experiment ["experimentTypeId" ] == constants .ExperimentType .SINGLE_NODE :
139139 data = (
140140 ("Name" , experiment ["templateHistory" ]["params" ].get ("name" )),
141- ("Handle " , experiment .get ("handle" )),
141+ ("ID " , experiment .get ("handle" )),
142142 ("State" , constants .ExperimentState .get_state_str (experiment .get ("state" ))),
143143 ("Ports" , experiment ["templateHistory" ]["params" ].get ("ports" )),
144- ("Project Handle " , experiment ["templateHistory" ]["params" ].get ("project_handle" )),
144+ ("Project ID " , experiment ["templateHistory" ]["params" ].get ("project_handle" )),
145145 ("Worker Command" , experiment ["templateHistory" ]["params" ].get ("worker_command" )),
146146 ("Worker Container" , experiment ["templateHistory" ]["params" ].get ("worker_container" )),
147147 ("Worker Machine Type" , experiment ["templateHistory" ]["params" ].get ("worker_machine_type" )),
@@ -154,7 +154,7 @@ def _make_details_table(experiment):
154154 constants .ExperimentType .MPI_MULTI_NODE ):
155155 data = (
156156 ("Name" , experiment ["templateHistory" ]["params" ].get ("name" )),
157- ("Handle " , experiment .get ("handle" )),
157+ ("ID " , experiment .get ("handle" )),
158158 ("State" , constants .ExperimentState .get_state_str (experiment .get ("state" ))),
159159 ("Artifact directory" , experiment ["templateHistory" ]["params" ].get ("artifactDirectory" )),
160160 ("Cluster ID" , experiment ["templateHistory" ]["params" ].get ("clusterId" )),
@@ -169,7 +169,7 @@ def _make_details_table(experiment):
169169 ("Parameter Server Machine Type" ,
170170 experiment ["templateHistory" ]["params" ].get ("parameter_server_machine_type" )),
171171 ("Ports" , experiment ["templateHistory" ]["params" ].get ("ports" )),
172- ("Project Handle " , experiment ["templateHistory" ]["params" ].get ("project_handle" )),
172+ ("Project ID " , experiment ["templateHistory" ]["params" ].get ("project_handle" )),
173173 ("Worker Command" , experiment ["templateHistory" ]["params" ].get ("worker_command" )),
174174 ("Worker Container" , experiment ["templateHistory" ]["params" ].get ("worker_container" )),
175175 ("Worker Count" , experiment ["templateHistory" ]["params" ].get ("worker_count" )),
@@ -185,8 +185,8 @@ def _make_details_table(experiment):
185185 return table_string
186186
187187
188- def get_experiment_details (experiment_handle , api = experiments_api ):
189- url = "/experiments/{}/" .format (experiment_handle )
188+ def get_experiment_details (experiment_id , api = experiments_api ):
189+ url = "/experiments/{}/" .format (experiment_id )
190190 response = api .get (url )
191191 details = response .content
192192 if response .ok :
0 commit comments