Skip to content

Commit 5424120

Browse files
committed
added filter
1 parent 6dce274 commit 5424120

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

openml/setups/functions.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def get_setup(setup_id):
7474
return _create_setup_from_xml(result_dict)
7575

7676

77-
def list_setups(flow=None, tag=None, offset=None, size=None):
77+
def list_setups(flow=None, tag=None, setup=None, offset=None, size=None):
7878
"""List all setups matching all of the given filters.
7979
8080
Perform API call `/setup/list/{filters}
@@ -85,6 +85,8 @@ def list_setups(flow=None, tag=None, offset=None, size=None):
8585
8686
tag : str, optional
8787
88+
setup : list(int), optional
89+
8890
offset : int, optional
8991
9092
size : int, optional
@@ -100,6 +102,8 @@ def list_setups(flow=None, tag=None, offset=None, size=None):
100102
api_call += "/offset/%d" % int(offset)
101103
if size is not None:
102104
api_call += "/limit/%d" % int(size)
105+
if size is not None:
106+
api_call += "/setup/%s" % ','.join([str(int(i)) for i in setup])
103107
if flow is not None:
104108
api_call += "/flow/%s" % flow
105109
if tag is not None:

0 commit comments

Comments
 (0)