Skip to content

Commit ca6e852

Browse files
committed
Adding __str__ for OpenMLStudy + Fixing flake issues
1 parent 837cb9b commit ca6e852

5 files changed

Lines changed: 63 additions & 7 deletions

File tree

openml/datasets/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ def __str__(self):
189189
num_instances = ''
190190
if object_dict['qualities']['NumberOfInstances'] is not None:
191191
num_instances = '%-14s: %d\n' % ("# of instances",
192-
object_dict['qualities']['NumberOfInstances'])
192+
object_dict['qualities']['NumberOfInstances'])
193193
num_features = '%-14s: %d\n' % ("# of features", len(object_dict['features']))
194194
output_str = name + version + format + date + licence + d_url + w_url + local_file + \
195195
pickle_file + num_instances + num_features

openml/evaluations/evaluation.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -55,23 +55,23 @@ def __str__(self):
5555
upload = '\n%-15s: %s\n\n' % ('Upload Date', object_dict['upload_time'])
5656
run = '%-15s: %d\n' % ('Run ID', object_dict['run_id'])
5757
run = run + '%-15s: %s\n\n' % ('OpenML Run URL',
58-
base_url + 'r/' + str(object_dict['run_id']))
58+
base_url + 'r/' + str(object_dict['run_id']))
5959

6060
task = '%-15s: %d\n' % ('Task ID', object_dict['task_id'])
6161
task = task + '%-15s: %s\n\n' % ('OpenML Task URL',
62-
base_url + 't/' + str(object_dict['task_id']))
62+
base_url + 't/' + str(object_dict['task_id']))
6363

6464
flow = '%-15s: %d\n' % ('Flow ID', object_dict['flow_id'])
6565
flow = flow + '%-15s: %s\n' % ('Flow Name', object_dict['flow_name'])
6666
flow = flow + '%-15s: %s\n\n' % ('OpenML Flow URL',
67-
base_url + 'f/' + str(object_dict['flow_id']))
67+
base_url + 'f/' + str(object_dict['flow_id']))
6868

6969
setup = '%-15s: %d\n\n' % ('Setup ID', object_dict['setup_id'])
7070

7171
data = '%-15s: %d\n' % ('Data ID', int(object_dict['data_id']))
7272
data = data + '%-15s: %s\n' % ('Data Name', object_dict['data_name'])
7373
data = data + '%-15s: %s\n\n' % ('OpenML Data URL',
74-
base_url + 'd/' + str(object_dict['data_id']))
74+
base_url + 'd/' + str(object_dict['data_id']))
7575

7676
metric = '%-15s: %s\n' % ('Metric Used', object_dict['function'])
7777
value = '%-15s: %f\n' % ('Result', object_dict['value'])

openml/flows/flow.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,8 @@ def __str__(self):
137137
output_str = ''
138138
id = '\n%-16s: %s\n' % ('Flow ID', object_dict['flow_id'])
139139
version = '%-16s: %s\n' % ('Flow Version', object_dict['version'])
140-
url = '%-16s: %s\n' % ('Flow URL', 'https://www.openml.org/f/' + str(object_dict['flow_id']))
140+
url = '%-16s: %s\n' % ('Flow URL',
141+
'https://www.openml.org/f/' + str(object_dict['flow_id']))
141142
name = '%-16s: %s\n' % ('Flow Name', object_dict['name'])
142143
description = '%-16s: %s\n\n' % ('Flow Description', object_dict['description'])
143144

openml/setups/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def __str__(self):
8181
flow = flow + '%-18s: %s\n' % ("Flow Full Name", object_dict['full_name'])
8282
url = 'https://www.openml.org/f/' + str(object_dict['flow_id'])
8383
flow = flow + '%-18s: %s\n' % ("Flow URL", url)
84-
filler = " "*4
84+
filler = " " * 4
8585
params = '%-18s: %s\n' % ("Parameter Name", object_dict['parameter_name'])
8686
params = params + filler + '%-14s: %s\n' % ("Data_Type", object_dict['data_type'])
8787
params = params + filler + '%-14s: %s\n' % ("Default", object_dict['default_value'])

openml/study/study.py

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,36 @@ def __init__(
8989
self.runs = runs
9090
pass
9191

92+
def __str__(self):
93+
object_dict = self.__dict__
94+
output_str = ''
95+
id = '\n%-16s: %s\n' % ("ID", object_dict['id'])
96+
name = '%-16s: %s\n' % ("Name", object_dict['name'])
97+
status = '%-16s: %s\n' % ("Status", object_dict['status'])
98+
main_entity_type = '%-16s: %s\n' % ("Main Entity Type", object_dict['main_entity_type'])
99+
url = 'https://www.openml.org/s/' + str(object_dict['id'])
100+
study_url = '%-16s: %s\n' % ("Study URL", url)
101+
data = ''
102+
if object_dict['data'] is not None:
103+
data = '%-16s: %s\n' % ("# of Data", len(object_dict['data']))
104+
tasks = ''
105+
if object_dict['tasks'] is not None:
106+
tasks = '%-16s: %s\n' % ("# of Tasks", len(object_dict['tasks']))
107+
flows = ''
108+
if object_dict['flows'] is not None:
109+
flows = '%-16s: %s\n' % ("# of Flows", len(object_dict['flows']))
110+
runs = ''
111+
if object_dict['runs'] is not None:
112+
runs = '%-16s: %s\n' % ("# of Runs", len(object_dict['runs']))
113+
114+
url = 'https://www.openml.org/u/' + str(object_dict['creator'])
115+
creator = '\n%-16s: %s\n' % ("Creator", url)
116+
upload_time = '%-16s: %s\n' % ("Upload Time",
117+
object_dict['creation_date'].replace('T', ' '))
118+
output_str = id + name + status + main_entity_type + study_url + data + \
119+
tasks + flows + runs + creator + upload_time
120+
return(output_str)
121+
92122
def publish(self) -> int:
93123
"""
94124
Publish the study on the OpenML server.
@@ -233,6 +263,31 @@ def __init__(
233263
setups=setups,
234264
)
235265

266+
# def __str__(self):
267+
# object_dict = self.__dict__
268+
# output_str = ''
269+
# id = '\n%-16s: %s\n' % ("ID", object_dict['id'])
270+
# name = '%-16s: %s\n' % ("Name", object_dict['name'])
271+
# status = '%-16s: %s\n' % ("Status", object_dict['status'])
272+
# main_entity_type = '%-16s: %s\n' % ("Main Entity Type", object_dict['main_entity_type'])
273+
# url = 'https://www.openml.org/s/' + str(object_dict['id'])
274+
# url = '%-16s: %s\n' % ("Study URL", url)
275+
# data = ''
276+
# if object_dict['data'] is not None:
277+
# data = '%-16s: %s\n' % ("# of Data", len(object_dict['data']))
278+
# tasks = ''
279+
# if object_dict['tasks'] is not None:
280+
# tasks = '%-16s: %s\n' % ("# of Tasks", len(object_dict['tasks']))
281+
# flows = ''
282+
# if object_dict['flows'] is not None:
283+
# flows = '%-16s: %s\n' % ("# of Flows", len(object_dict['flows']))
284+
# runs = ''
285+
# if object_dict['runs'] is not None:
286+
# runs = '%-16s: %s\n' % ("# of Runs", len(object_dict['runs']))
287+
# output_str = id + name + status + main_entity_type + url + data + \
288+
# tasks + flows + runs
289+
# return(output_str)
290+
236291

237292
class OpenMLBenchmarkSuite(BaseStudy):
238293

0 commit comments

Comments
 (0)