Skip to content

Commit 23aaf81

Browse files
committed
changed comments of setup, changed assertions
1 parent 70475fe commit 23aaf81

2 files changed

Lines changed: 17 additions & 6 deletions

File tree

openml/setups/setup.py

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,21 @@ class OpenMLParameter(object):
2020
2121
Parameters
2222
----------
23-
flow_id : int
24-
The flow that it is build upon
25-
parameters : dict
26-
The setting of the parameters
23+
id : int
24+
The input id from the openml database
25+
flow id : int
26+
The flow to which this parameter is associated
27+
full_name : str
28+
The name of the flow and parameter combined
29+
parameter_name : str
30+
The name of the parameter
31+
data_type : str
32+
The datatype of the parameter. generally unused for sklearn flows
33+
default_value : str
34+
The default value. For sklearn parameters, this is unknown and a
35+
default value is selected arbitrarily
36+
value : str
37+
If the parameter was set, the value that it was set to.
2738
"""
2839
def __init__(self, id, flow_id, full_name, parameter_name, data_type, default_value, value):
2940
self.id = id

tests/test_setups/test_setup_functions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,6 @@ def test_setup_get(self):
8787
current = openml.setups.get_setup(setups[idx])
8888
assert current.flow_id > 0
8989
if num_params[idx] == 0:
90-
assert current.parameters is None
90+
self.asserts(current.parameters is None)
9191
else:
92-
assert len(current.parameters) == num_params[idx]
92+
self.asserts(len(current.parameters) == num_params[idx])

0 commit comments

Comments
 (0)