adjusting to new changes

openml · satvshr · Feb 3, 2026 · Feb 3, 2026 · Feb 3, 2026 · Feb 3, 2026
commit 1877c073063af4fe2b9a4d01316a2dae241f9a3d
diff --git a/openml/_api/resources/task.py b/openml/_api/resources/task.py
@@ -40,7 +40,7 @@ def get(self, task_id: int) -> OpenMLTask:
         if not isinstance(task_id, int):
             raise TypeError(f"Task id should be integer, is {type(task_id)}")
 
-        response = self._http.get(f"task/{task_id}")
+        response = self._http.get(f"task/{task_id}", use_cache=True)
         return self._create_task_from_xml(response.text)
 
     def _create_task_from_xml(self, xml: str) -> OpenMLTask:
@@ -330,7 +330,7 @@ def _get_estimation_procedure_list(self) -> builtins.list[dict[str, Any]]:
 
 class TaskV2API(ResourceV2API, TaskAPI):
     def get(self, task_id: int) -> OpenMLTask:
-        response = self._http.get(f"tasks/{task_id}")
+        response = self._http.get(f"tasks/{task_id}", use_cache=True)
         return self._create_task_from_json(response.json())
 
     def _create_task_from_json(self, task_json: dict) -> OpenMLTask:
@@ -380,4 +380,4 @@ def list(
         task_type: TaskType | int | None = None,
         **kwargs: Any,
     ) -> pd.DataFrame:
-        raise NotImplementedError("Task listing is not available in API v2 yet.")
+        raise NotImplementedError(self._not_supported(method="list"))
diff --git a/openml/tasks/functions.py b/openml/tasks/functions.py
@@ -140,6 +140,7 @@ def get_tasks(
     return tasks
 
 
+@openml.utils.thread_safe_if_oslo_installed
 def get_task(
     task_id: int,
     download_splits: bool = False,  # noqa: FBT002
@@ -241,7 +242,13 @@ def create_task(
     elif task_type == TaskType.SUPERVISED_REGRESSION:
         task_cls = OpenMLRegressionTask  # type: ignore
     else:
-        raise NotImplementedError(f"Task type {task_type:d} not supported.")
+        raise NotImplementedError(
+            f"Task type ID {task_type:d} is not supported. "
+            f"Supported task type IDs: {TaskType.SUPERVISED_CLASSIFICATION.value},"
+            f"{TaskType.SUPERVISED_REGRESSION.value}, "
+            f"{TaskType.CLUSTERING.value}, {TaskType.LEARNING_CURVE.value}. "
+            f"Please refer to the TaskType enum for valid task type identifiers."
+        )
 
     return task_cls(
         task_type_id=task_type,
@@ -270,4 +277,4 @@ def delete_task(task_id: int) -> bool:
     bool
         True if the deletion was successful. False otherwise.
     """
-    return openml.utils._delete_entity("task", task_id)
+    return openml._backend.task.delete(task_id)
diff --git a/openml/tasks/task.py b/openml/tasks/task.py
@@ -13,7 +13,6 @@
 
 import openml.config
 from openml import datasets
-from openml._api.resources.base import ResourceAPI
 from openml.base import OpenMLBase
 from openml.utils import _create_cache_directory_for_id
 
@@ -46,7 +45,7 @@ class _EstimationProcedure(TypedDict):
     data_splits_url: str | None
 
 
-class OpenMLTask(OpenMLBase, ResourceAPI):
+class OpenMLTask(OpenMLBase):
     """OpenML Task object.
 
     Parameters
@@ -219,6 +218,47 @@ def _parse_publish_response(self, xml_response: dict) -> None:
         """Parse the id from the xml_response and assign it to self."""
         self.task_id = int(xml_response["oml:upload_task"]["oml:id"])
 
+    def publish(self) -> OpenMLTask:
+        """Publish this task to OpenML server.
+
+        Returns
+        -------
+        self : OpenMLTask
+        """
+        file_elements = self._get_file_elements()
+        if "description" not in file_elements:
+            file_elements["description"] = self._to_xml()
+        task_id = openml._backend.task.publish(path="task", files=file_elements)
+        self.task_id = task_id
+        return self
+
+    def push_tag(self, tag: str) -> None:
+        """Annotates this task with a tag on the server.
+
+        Parameters
+        ----------
+        tag : str
+            Tag to attach to the task.
+        """
+        if self.task_id is None:
+            raise ValueError(
+                "Task does not have an ID. Please publish the task before tagging."
+            )
+        openml._backend.task.tag(self.task_id, tag)
+
+    def remove_tag(self, tag: str) -> None:
+        """Removes a tag from this task on the server.
+
+        Parameters
+        ----------
+        tag : str
+            Tag to remove from the task.
+        """
+        if self.task_id is None:
+            raise ValueError(
+                "Dataset does not have an ID. Please publish the dataset before untagging."
+            )
+        openml._backend.task.untag(self.task_id, tag)
 
 class OpenMLSupervisedTask(OpenMLTask, ABC):
     """OpenML Supervised Classification object.