@@ -33,15 +33,15 @@ def setup_v2(http_client_v2, minio_client) -> SetupV2API:
3333 return SetupV2API (http = http_client_v2 , minio = minio_client )
3434
3535
36- @pytest .mark .uses_test_server ()
36+ @pytest .mark .test_server ()
3737def test_v1_list (setup_v1 ):
3838 setups = setup_v1 .list (limit = 10 , offset = 0 )
3939
4040 assert isinstance (setups , list )
4141 assert len (setups ) > 0
4242 assert all (isinstance (s , OpenMLSetup ) for s in setups )
4343
44- @pytest .mark .uses_test_server ()
44+ @pytest .mark .test_server ()
4545def test_v1_get (setup_v1 ):
4646 setup_id = 1
4747 setup = setup_v1 .get (setup_id )
@@ -50,7 +50,7 @@ def test_v1_get(setup_v1):
5050 assert setup .setup_id == setup_id
5151
5252@pytest .mark .sklearn ()
53- @pytest .mark .uses_test_server ()
53+ @pytest .mark .test_server ()
5454def test_v1_exists_nonexisting_setup (setup_v1 ):
5555 """Test exists() returns False when setup doesn't exist"""
5656 # first publish a non-existing flow
@@ -70,7 +70,7 @@ def test_v1_exists_nonexisting_setup(setup_v1):
7070 assert not setup_id
7171
7272@pytest .mark .sklearn ()
73- @pytest .mark .uses_test_server ()
73+ @pytest .mark .test_server ()
7474def test_v1_exists_existing_setup (setup_v1 ):
7575 """Test exists() returns setup_id when setup exists"""
7676 flow = SklearnExtension ().model_to_flow (
@@ -95,18 +95,18 @@ def test_v1_exists_existing_setup(setup_v1):
9595 setup_id = setup_v1 .exists (flow , openml_param_settings )
9696 assert setup_id == run .setup_id
9797
98- @pytest .mark .uses_test_server ()
98+ @pytest .mark .test_server ()
9999def test_v2_list (setup_v2 ):
100100 with pytest .raises (OpenMLNotSupportedError ):
101101 setup_v2 .list (limit = 10 , offset = 0 )
102102
103- @pytest .mark .uses_test_server ()
103+ @pytest .mark .test_server ()
104104def test_v2_get (setup_v2 ):
105105 with pytest .raises (OpenMLNotSupportedError ):
106106 setup_v2 .get (1 )
107107
108108@pytest .mark .sklearn ()
109- @pytest .mark .uses_test_server ()
109+ @pytest .mark .test_server ()
110110def test_v2_exists_nonexisting_setup (setup_v2 ):
111111 # first publish a non-existing flow
112112 sentinel = get_sentinel ()
@@ -126,7 +126,7 @@ def test_v2_exists_nonexisting_setup(setup_v2):
126126
127127
128128@pytest .mark .sklearn ()
129- @pytest .mark .uses_test_server ()
129+ @pytest .mark .test_server ()
130130def test_v2_exists_existing_setup (setup_v2 ):
131131 flow = SklearnExtension ().model_to_flow (
132132 sklearn .naive_bayes .GaussianNB ()
0 commit comments