module level import for tests

Author: geetu040

openml/_api/__init__.py

@@ -5,24 +5,35 @@
 )
 from .resources import (
     API_REGISTRY,
+    DatasetAPI,
     DatasetV1API,
     DatasetV2API,
+    EstimationProcedureAPI,
     EstimationProcedureV1API,
     EstimationProcedureV2API,
+    EvaluationAPI,
+    EvaluationMeasureAPI,
     EvaluationMeasureV1API,
     EvaluationMeasureV2API,
     EvaluationV1API,
     EvaluationV2API,
     FallbackProxy,
+    FlowAPI,
     FlowV1API,
     FlowV2API,
     ResourceAPI,
+    ResourceV1API,
+    ResourceV2API,
+    RunAPI,
     RunV1API,
     RunV2API,
+    SetupAPI,
     SetupV1API,
     SetupV2API,
+    StudyAPI,
     StudyV1API,
     StudyV2API,
+    TaskAPI,
     TaskV1API,
     TaskV2API,
 )
@@ -44,27 +55,40 @@
     "CacheConfig",
     "Config",
     "ConnectionConfig",
+    "DatasetAPI",
     "DatasetV1API",
     "DatasetV2API",
+    "EstimationProcedureAPI",
     "EstimationProcedureV1API",
     "EstimationProcedureV2API",
+    "EvaluationAPI",
+    "EvaluationMeasureAPI",
     "EvaluationMeasureV1API",
     "EvaluationMeasureV2API",
     "EvaluationV1API",
     "EvaluationV2API",
     "FallbackProxy",
+    "FallbackProxy",
+    "FlowAPI",
     "FlowV1API",
     "FlowV2API",
     "HTTPCache",
     "HTTPClient",
     "MinIOClient",
     "ResourceAPI",
+    "ResourceAPI",
+    "ResourceV1API",
+    "ResourceV2API",
+    "RunAPI",
     "RunV1API",
     "RunV2API",
+    "SetupAPI",
     "SetupV1API",
     "SetupV2API",
+    "StudyAPI",
     "StudyV1API",
     "StudyV2API",
+    "TaskAPI",
     "TaskV1API",
     "TaskV2API",
     "_backend",

openml/_api/resources/__init__.py

@@ -1,5 +1,19 @@
 from ._registry import API_REGISTRY
-from .base import FallbackProxy, ResourceAPI
+from .base import (
+    DatasetAPI,
+    EstimationProcedureAPI,
+    EvaluationAPI,
+    EvaluationMeasureAPI,
+    FallbackProxy,
+    FlowAPI,
+    ResourceAPI,
+    ResourceV1API,
+    ResourceV2API,
+    RunAPI,
+    SetupAPI,
+    StudyAPI,
+    TaskAPI,
+)
 from .dataset import DatasetV1API, DatasetV2API
 from .estimation_procedure import (
     EstimationProcedureV1API,
@@ -15,24 +29,37 @@
 
 __all__ = [
     "API_REGISTRY",
+    "DatasetAPI",
     "DatasetV1API",
     "DatasetV2API",
+    "EstimationProcedureAPI",
     "EstimationProcedureV1API",
     "EstimationProcedureV2API",
+    "EvaluationAPI",
+    "EvaluationMeasureAPI",
     "EvaluationMeasureV1API",
     "EvaluationMeasureV2API",
     "EvaluationV1API",
     "EvaluationV2API",
     "FallbackProxy",
+    "FallbackProxy",
+    "FlowAPI",
     "FlowV1API",
     "FlowV2API",
     "ResourceAPI",
+    "ResourceAPI",
+    "ResourceV1API",
+    "ResourceV2API",
+    "RunAPI",
     "RunV1API",
     "RunV2API",
+    "SetupAPI",
     "SetupV1API",
     "SetupV2API",
+    "StudyAPI",
     "StudyV1API",
     "StudyV2API",
+    "TaskAPI",
     "TaskV1API",
     "TaskV2API",
 ]

openml/testing.py

@@ -16,7 +16,7 @@
 import requests
 
 import openml
-from openml._api.clients import HTTPCache, HTTPClient
+from openml._api import HTTPCache, HTTPClient
 from openml.enums import RetryPolicy
 from openml.exceptions import OpenMLServerException
 from openml.tasks import TaskType

tests/test_api/test_versions.py

@@ -1,6 +1,6 @@
 import pytest
 from openml.testing import TestAPIBase
-from openml._api.resources.base.versions import ResourceV1API
+from openml._api import ResourceV1API
 from openml.enums import ResourceType