added tag and untag methods

Signed-off-by: Omswastik-11 <omswastikpanda11@gmail.com>

Author: Omswastik-11

openml/_api/resources/flows.py

@@ -152,16 +152,6 @@ def publish(self, path: str | None = None, files: Mapping[str, Any] | None = Non
         path = "flow"
         return super().publish(path, files)
 
-    def delete(self, flow_id: int) -> bool:
-        """Delete a flow from the OpenML server.
-
-        Parameters
-        ----------
-        flow_id : int
-            The ID of the flow to delete.
-        """
-        return super().delete(flow_id)
-
 
 class FlowsV2(ResourceV2, FlowsAPI):
     def get(
@@ -221,18 +211,15 @@ def exists(self, name: str, external_version: str) -> int | bool:
     def list(
         self,
         *,
-        limit: int | None = None,
-        offset: int | None = None,
-        tag: str | None = None,
-        uploader: str | None = None,
+        limit: int | None = None,  # noqa: ARG002
+        offset: int | None = None,  # noqa: ARG002
+        tag: str | None = None,  # noqa: ARG002
+        uploader: str | None = None,  # noqa: ARG002
     ) -> pd.DataFrame:
-        raise NotImplementedError("flows (list) not yet implemented in v2 server")
-
-    def publish(self, flow: OpenMLFlow) -> OpenMLFlow:  # type: ignore[override]
-        raise NotImplementedError("POST /flows (create) not yet implemented in v2 server")
+        self._not_supported(method="list")
 
-    def delete(self, flow_id: int) -> bool:
-        raise NotImplementedError("DELETE /flows/{id} not yet implemented in v2 server")
+    def publish(self, path: str | None = None, files: Mapping[str, Any] | None = None) -> int:  # type: ignore[override]  # noqa: ARG002
+        self._not_supported(method="publish")
 
     @staticmethod
     def _convert_v2_to_v1_format(v2_json: dict[str, Any]) -> dict[str, dict]:

openml/flows/flow.py

@@ -444,6 +444,7 @@ def publish(self, raise_error_if_exists: bool = False) -> OpenMLFlow:  # noqa: F
             if "description" not in file_elements:
                 file_elements["description"] = self._to_xml()
 
+            # Use api_context.backend.flows.publish which internally calls ResourceV1.publish
             flow_id = api_context.backend.flows.publish(path="flow", files=file_elements)
             self.flow_id = flow_id
         elif raise_error_if_exists:
@@ -473,6 +474,34 @@ def publish(self, raise_error_if_exists: bool = False) -> OpenMLFlow:  # noqa: F
             ) from e
         return self
 
+    def push_tag(self, tag: str) -> None:
+        """Annotates this flow with a tag on the server.
+
+        Parameters
+        ----------
+        tag : str
+            Tag to attach to the flow.
+        """
+        from openml._api import api_context
+
+        if self.flow_id is None:
+            raise ValueError("Flow does not have an ID. Please publish the flow before tagging.")
+        api_context.backend.flows.tag(self.flow_id, tag)
+
+    def remove_tag(self, tag: str) -> None:
+        """Removes a tag from this flow on the server.
+
+        Parameters
+        ----------
+        tag : str
+            Tag to remove from the flow.
+        """
+        from openml._api import api_context
+
+        if self.flow_id is None:
+            raise ValueError("Flow does not have an ID. Please publish the flow before untagging.")
+        api_context.backend.flows.untag(self.flow_id, tag)
+
     def get_structure(self, key_item: str) -> dict[str, list[str]]:
         """
         Returns for each sub-component of the flow the path of identifiers

openml/flows/functions.py

@@ -69,7 +69,7 @@ def _get_cached_flow(fid: int) -> OpenMLFlow:
         raise OpenMLCacheException(f"Flow file for fid {fid} not cached") from e
 
 
-# @openml.utils.thread_safe_if_oslo_installed
+@openml.utils.thread_safe_if_oslo_installed
 def get_flow(flow_id: int, reinstantiate: bool = False, strict_version: bool = True) -> OpenMLFlow:  # noqa: FBT002
     """Download the OpenML flow for a given flow ID.