fix tests

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

Author: Omswastik-11

tests/test_api/test_flow.py

@@ -20,7 +20,12 @@ def flow_v1(http_client_v1, minio_client) -> FlowV1API:
 
 
 @pytest.fixture
-def flow_v2(http_client_v2, minio_client) -> FlowV2API:
+def flow_v2(minio_client) -> FlowV2API:
+    from openml.enums import APIVersion
+    if openml.config.servers[APIVersion.V2]["server"] is None:
+        pytest.skip("V2 server is not configured")
+    from openml._api import HTTPClient
+    http_client_v2 = HTTPClient(api_version=APIVersion.V2)
     return FlowV2API(http=http_client_v2, minio=minio_client)
 
 

tests/test_flows/test_flow.py

@@ -7,11 +7,11 @@
 import re
 import os
 import time
-import uuid
 from packaging.version import Version
 from unittest import mock
 
 import pytest
+import requests
 import scipy.stats
 import sklearn
 import sklearn.datasets
@@ -31,8 +31,8 @@
 import openml
 import openml.exceptions
 import openml.utils
-import requests
-from openml.testing import SimpleImputer, TestBase, create_request_response
+
+from openml.testing import SimpleImputer, TestBase
 
 
 class TestFlow(TestBase):
@@ -117,7 +117,7 @@ def test_tagging(self):
         flow.push_tag(tag)
         flows = openml.flows.list_flows(tag=tag)
         assert len(flows) == 1
-        assert flow_id in flows["id"].values
+        assert flow_id in flows["id"]
         flow.remove_tag(tag)
         flows = openml.flows.list_flows(tag=tag)
         assert len(flows) == 0
@@ -299,15 +299,14 @@ def test_semi_legal_flow(self):
         TestBase._mark_entity_for_removal("flow", flow.flow_id, flow.name)
         TestBase.logger.info(f"collected from {__file__.split('/')[-1]}: {flow.flow_id}")
 
-    
     @pytest.mark.sklearn()
     @mock.patch("openml.flows.functions.get_flow")
     @mock.patch("openml.flows.functions.flow_exists")
     @mock.patch("requests.Session.request")
     def test_publish_error(self, mock_request, flow_exists_mock, get_flow_mock):
         model = sklearn.ensemble.RandomForestClassifier()
         flow = self.extension.model_to_flow(model)
-
+        
         # Create mock response directly
         response = requests.Response()
         response.status_code = 200