lazy imports

Author: JATAYU000

openml/datasets/functions.py

@@ -17,7 +17,6 @@
 
 import openml._api_calls
 import openml.utils
-from openml._api import api_context
 from openml.exceptions import (
     OpenMLServerError,
 )
@@ -57,6 +56,8 @@ def list_qualities() -> list[str]:
     -------
     list
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.list_qualities()
 
 
@@ -110,6 +111,8 @@ def list_datasets(
         If qualities are calculated for the dataset, some of
         these are also included as columns.
     """
+    from openml._api import api_context
+
     listing_call = partial(
         api_context.backend.datasets.list,
         data_id=data_id,
@@ -344,6 +347,8 @@ def get_dataset(
     dataset : :class:`openml.OpenMLDataset`
         The downloaded dataset.
     """
+    from openml._api import api_context
+
     if download_all_files:
         warnings.warn(
             "``download_all_files`` is experimental and is likely to break with new releases.",
@@ -654,6 +659,8 @@ def status_update(data_id: int, status: Literal["active", "deactivated"]) -> Non
     status : str,
         'active' or 'deactivated'
     """
+    from openml._api import api_context
+
     legal_status = {"active", "deactivated"}
     if status not in legal_status:
         raise ValueError(f"Illegal status value. Legal values: {legal_status}")
@@ -732,6 +739,8 @@ def edit_dataset(
     -------
     Dataset id
     """
+    from openml._api import api_context
+
     if not isinstance(data_id, int):
         raise TypeError(f"`data_id` must be of type `int`, not {type(data_id)}.")
 
@@ -780,6 +789,8 @@ def fork_dataset(data_id: int) -> int:
     Dataset id of the forked dataset
 
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.fork(dataset_id=data_id)
 
 
@@ -804,6 +815,8 @@ def data_feature_add_ontology(data_id: int, index: int, ontology: str) -> bool:
     -------
     True or throws an OpenML server exception
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.feature_add_ontology(data_id, index, ontology)
 
 
@@ -827,6 +840,8 @@ def data_feature_remove_ontology(data_id: int, index: int, ontology: str) -> boo
     -------
     True or throws an OpenML server exception
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.feature_remove_ontology(data_id, index, ontology)
 
 
@@ -847,6 +862,8 @@ def _topic_add_dataset(data_id: int, topic: str) -> int:
     -------
     Dataset id
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.add_topic(data_id, topic)
 
 
@@ -867,6 +884,8 @@ def _topic_delete_dataset(data_id: int, topic: str) -> int:
     -------
     Dataset id
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.delete_topic(data_id, topic)
 
 
@@ -947,6 +966,8 @@ def _get_dataset_parquet(
     output_filename : Path, optional
         Location of the Parquet file if successfully downloaded, None otherwise.
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.download_dataset_parquet(description, download_all_files)
 
 
@@ -977,6 +998,8 @@ def _get_dataset_arff(
     output_filename : Path
         Location of ARFF file.
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.download_dataset_arff(description)
 
 
@@ -1005,7 +1028,7 @@ def _get_dataset_features_file(
     Path
         Path of the cached dataset feature file
     """
-    # cache directory not used here anymore
+    from openml._api import api_context
 
     return api_context.backend.datasets.download_features_file(dataset_id)
 
@@ -1035,7 +1058,7 @@ def _get_dataset_qualities_file(
     str
         Path of the cached qualities file
     """
-    # cache directory not used here anymore
+    from openml._api import api_context
 
     return api_context.backend.datasets.download_qualities_file(dataset_id)
 
@@ -1055,6 +1078,8 @@ def _get_online_dataset_arff(dataset_id: int) -> str | None:
     str or None
         A string representation of an ARFF file. Or None if file already exists.
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.get_online_dataset_arff(dataset_id)
 
 
@@ -1072,6 +1097,8 @@ def _get_online_dataset_format(dataset_id: int) -> str:
     str
         Dataset format.
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.get_online_dataset_format(dataset_id)
 
 
@@ -1091,4 +1118,6 @@ def delete_dataset(dataset_id: int) -> bool:
     bool
         True if the deletion was successful. False otherwise.
     """
+    from openml._api import api_context
+
     return api_context.backend.datasets.delete(dataset_id)