def test_creation_and_hydration(self):
run_data, metrics, params = TestRunData._create()
(run_info, run_uuid, experiment_id, name, source_type, source_name, entry_point_name,
user_id, status, start_time, end_time, source_version, tags,
artifact_uri) = TestRunInfo._create()
run1 = Run(run_info, run_data)
self._check_run(run1, run_info, run_data)
as_dict = {"info": {"run_uuid": run_uuid,
"experiment_id": experiment_id,
"name": name,
"source_type": source_type,
"source_name": source_name,
"entry_point_name": entry_point_name,
"user_id": user_id,
"status": status,
"start_time": start_time,
"end_time": end_time,
"source_version": source_version,
"tags": tags,
"artifact_uri": artifact_uri,
},
"data": {"metrics": metrics,
"params": params}}
self.assertEqual(dict(run1), as_dict)
# proto = run1.to_proto()
# run2 = Run.from_proto(proto)
# self._check_run(run2, run_info, run_data)
run3 = Run.from_dictionary(as_dict)
self._check_run(run3, run_info, run_data)
def test_creation_and_hydration(self):
run_data, metrics, params, tags = TestRunData._create()
(run_info, run_uuid, experiment_id, name, source_type, source_name,
entry_point_name, user_id, status, start_time, end_time,
source_version, lifecycle_stage,
artifact_uri) = TestRunInfo._create()
run1 = Run(run_info, run_data)
self._check_run(run1, run_info, run_data)
as_dict = {
"info": {
"run_uuid": run_uuid,
"experiment_id": experiment_id,
"name": name,
"source_type": source_type,
"source_name": source_name,
"entry_point_name": entry_point_name,
"user_id": user_id,
"status": status,
"start_time": start_time,
"end_time": end_time,
"source_version": source_version,
"lifecycle_stage": lifecycle_stage,
"artifact_uri": artifact_uri,
},
"data": {
"metrics": [dict(m) for m in metrics],
"params": [dict(p) for p in params],
"tags": [dict(t) for t in tags]
}
}
self.assertEqual(run1.to_dictionary(), as_dict)
proto = run1.to_proto()
run2 = Run.from_proto(proto)
self._check_run(run2, run_info, run_data)
run3 = Run.from_dictionary(as_dict)
self._check_run(run3, run_info, run_data)
