Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
list evals - update
  • Loading branch information
Bilgecelik committed Jul 13, 2020
commit dca4d971342594d0798cd2d314c22256a71d7790
14 changes: 8 additions & 6 deletions openml/evaluations/functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,11 @@ def list_evaluations(
function: str,
offset: Optional[int] = None,
size: Optional[int] = None,
tasks: Optional[List[Union[str,int]]] = None,
setups: Optional[List[Union[str,int]]] = None,
flows: Optional[List[Union[str,int]]] = None,
runs: Optional[List[Union[str,int]]] = None,
uploaders: Optional[List[Union[str,int]]] = None,
tasks: Optional[List[Union[str, int]]] = None,
setups: Optional[List[Union[str, int]]] = None,
flows: Optional[List[Union[str, int]]] = None,
runs: Optional[List[Union[str, int]]] = None,
uploaders: Optional[List[Union[str, int]]] = None,
tag: Optional[str] = None,
study: Optional[int] = None,
per_fold: Optional[bool] = None,
Expand Down Expand Up @@ -363,7 +363,9 @@ def list_evaluations_setups(
)
setup_data = pd.DataFrame()
for setups in setup_chunks:
result = pd.DataFrame(openml.setups.list_setups(setup=setups, output_format="dataframe"))
result = pd.DataFrame(
openml.setups.list_setups(setup=setups, output_format="dataframe")
)
result.drop("flow_id", axis=1, inplace=True)
# concat resulting setup chunks into single datframe
setup_data = pd.concat([setup_data, result], ignore_index=True)
Expand Down