Skip to content

Commit 7ae536a

Browse files
committed
add http_client in TestBase
1 parent fe7cf96 commit 7ae536a

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

openml/testing.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@
1515
import requests
1616

1717
import openml
18+
from openml._api import HTTPClient
19+
from openml.enums import APIVersion
1820
from openml.exceptions import OpenMLServerException
1921
from openml.tasks import TaskType
2022

@@ -53,6 +55,8 @@ class TestBase(unittest.TestCase):
5355
logger = logging.getLogger("unit_tests_published_entities")
5456
logger.setLevel(logging.DEBUG)
5557

58+
http_client: HTTPClient = HTTPClient(api_version=APIVersion.V1)
59+
5660
def setUp(self, n_levels: int = 1, tmpdir_suffix: str = "") -> None:
5761
"""Setup variables and temporary directories.
5862

0 commit comments

Comments
 (0)