Skip to content

Commit 5d229c7

Browse files
committed
use minio in run migration
1 parent ea3a8ae commit 5d229c7

2 files changed

Lines changed: 12 additions & 6 deletions

File tree

openml/_api/resources/run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ def get(
4444
path = f"run/{run_id}"
4545
response = self._http.get(
4646
path,
47-
use_cache=True,
48-
reset_cache=reset_cache,
47+
enable_cache=True,
48+
refresh_cache=reset_cache,
4949
use_api_key=True,
5050
)
5151
xml_content = response.text

tests/test_api/test_run.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ class TestRunV1API(TestRunAPIBase):
6060
def setUp(self):
6161
super().setUp()
6262
http_client = self.http_clients[APIVersion.V1]
63-
self.resource = RunV1API(http_client)
63+
self.resource = RunV1API(http=http_client, minio=self.minio_client)
6464

6565
def test_get(self):
6666
self._get()
@@ -76,7 +76,7 @@ class TestRunV2API(TestRunAPIBase):
7676
def setUp(self):
7777
super().setUp()
7878
http_client = self.http_clients[APIVersion.V2]
79-
self.resource = RunV2API(http_client)
79+
self.resource = RunV2API(http=http_client, minio=self.minio_client)
8080

8181
def test_get(self):
8282
with pytest.raises(
@@ -103,8 +103,14 @@ def test_publish_and_delete(self):
103103
class TestRunCombinedAPI(TestAPIBase):
104104
def setUp(self):
105105
super().setUp()
106-
self.resource_v1 = RunV1API(self.http_clients[APIVersion.V1])
107-
self.resource_v2 = RunV2API(self.http_clients[APIVersion.V2])
106+
self.resource_v1 = RunV1API(
107+
http=self.http_clients[APIVersion.V1],
108+
minio=self.minio_client,
109+
)
110+
self.resource_v2 = RunV2API(
111+
http=self.http_clients[APIVersion.V2],
112+
minio=self.minio_client,
113+
)
108114

109115
def test_get_contracts(self):
110116
run_v1 = self.resource_v1.get(run_id=1)

0 commit comments

Comments
 (0)