updated the test

Author: Omswastik-11

openml/_api/resources/base/versions.py

@@ -2,6 +2,7 @@
 
 from collections.abc import Mapping
 from typing import Any, cast
+from xml.parsers.expat import ExpatError
 
 import xmltodict
 
@@ -71,7 +72,7 @@ def publish(self, path: str, files: Mapping[str, Any] | None) -> int:
             If the server returns an error during upload.
         """
         response = self._http.post(path, files=files)
-        parsed_response = xmltodict.parse(response.content)
+        parsed_response = self._parse_xml_response(response.content)
         return self._extract_id_from_upload(parsed_response)
 
     def delete(self, resource_id: int) -> bool:
@@ -106,7 +107,7 @@ def delete(self, resource_id: int) -> bool:
         path = f"{endpoint_name}/{resource_id}"
         try:
             response = self._http.delete(path)
-            result = xmltodict.parse(response.content)
+            result = self._parse_xml_response(response.content)
             return f"oml:{endpoint_name}_delete" in result
         except OpenMLServerException as e:
             self._handle_delete_exception(endpoint_name, e)
@@ -143,7 +144,7 @@ def tag(self, resource_id: int, tag: str) -> list[str]:
         data = {f"{endpoint_name}_id": resource_id, "tag": tag}
         response = self._http.post(path, data=data)
 
-        parsed_response = xmltodict.parse(response.content, force_list={"oml:tag"})
+        parsed_response = self._parse_xml_response(response.content, force_list={"oml:tag"})
         result = parsed_response[f"oml:{endpoint_name}_tag"]
         tags: list[str] = result.get("oml:tag", [])
 
@@ -180,12 +181,28 @@ def untag(self, resource_id: int, tag: str) -> list[str]:
         data = {f"{endpoint_name}_id": resource_id, "tag": tag}
         response = self._http.post(path, data=data)
 
-        parsed_response = xmltodict.parse(response.content, force_list={"oml:tag"})
+        parsed_response = self._parse_xml_response(response.content, force_list={"oml:tag"})
         result = parsed_response[f"oml:{endpoint_name}_untag"]
         tags: list[str] = result.get("oml:tag", [])
 
         return tags
 
+    def _parse_xml_response(self, payload: bytes | str, **kwargs: Any) -> Mapping[str, Any]:
+        try:
+            return cast("Mapping[str, Any]", xmltodict.parse(payload, **kwargs))
+        except ExpatError:
+            payload_text = (
+                payload.decode("utf-8", errors="ignore") if isinstance(payload, bytes) else payload
+            )
+            xml_start = payload_text.find("<?xml")
+            if xml_start == -1:
+                xml_start = payload_text.find("<oml:")
+            if xml_start == -1:
+                raise
+
+            xml_text = payload_text[xml_start:]
+            return cast("Mapping[str, Any]", xmltodict.parse(xml_text, **kwargs))
+
     def _get_endpoint_name(self) -> str:
         if self.resource_type == ResourceType.DATASET:
             return "data"