Skip to content

Commit 18d3439

Browse files
committed
reviews updated
1 parent 56ec10a commit 18d3439

3 files changed

Lines changed: 8 additions & 8 deletions

File tree

openml/_api/resources/base/resources.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,15 +6,15 @@
66
from pathlib import Path
77
from typing import TYPE_CHECKING, Any, Literal
88

9-
if TYPE_CHECKING:
10-
import pandas as pd
11-
12-
from openml.datasets.dataset import OpenMLDataFeature, OpenMLDataset
139
from openml.enums import ResourceType
1410

1511
from .base import ResourceAPI
1612

1713
if TYPE_CHECKING:
14+
import pandas as pd
15+
16+
from openml.datasets.data_feature import OpenMLDataFeature
17+
from openml.datasets.dataset import OpenMLDataset
1818
from openml.estimation_procedures import OpenMLEstimationProcedure
1919
from openml.evaluations import OpenMLEvaluation
2020
from openml.flows.flow import OpenMLFlow
@@ -88,12 +88,12 @@ def get_qualities(self, dataset_id: int) -> dict[str, float] | None: ...
8888

8989
@abstractmethod
9090
def parse_features_file(
91-
self, features_file: Path, features_pickle_file: Path
91+
self, features_file: Path, features_pickle_file: Path | None = None
9292
) -> dict[int, OpenMLDataFeature]: ...
9393

9494
@abstractmethod
9595
def parse_qualities_file(
96-
self, qualities_file: Path, qualities_pickle_file: Path
96+
self, qualities_file: Path, qualities_pickle_file: Path | None = None
9797
) -> dict[str, float]: ...
9898

9999
@abstractmethod

openml/_api/resources/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1505,7 +1505,7 @@ def get_online_dataset_format(self, dataset_id: int) -> str:
15051505
"""
15061506
dataset_json = self._http.get(f"datasets/{dataset_id}").json()
15071507
# build a dict from the json and get the format from the dataset description
1508-
return dataset_json["data_set_description"]["format"].lower() # type: ignore
1508+
return dataset_json["format"].lower() # type: ignore
15091509

15101510
def get_online_dataset_arff(self, dataset_id: int) -> str | None:
15111511
"""Download the ARFF file for a given dataset id

tests/test_runs/test_run_functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1969,7 +1969,7 @@ def test__run_task_get_arffcontent_2(parallel_mock):
19691969
]
19701970
)
19711971
@pytest.mark.test_server()
1972-
def test_joblib_backends(parallel_mock, n_jobs, backend, call_count, tmp_path):
1972+
def test_joblib_backends(parallel_mock, n_jobs, backend, call_count):
19731973
"""Tests evaluation of a run using various joblib backends and n_jobs."""
19741974

19751975
if backend is None:

0 commit comments

Comments
 (0)