From 67667f250b7ed0523e8d960a9f6061894f86ea1f Mon Sep 17 00:00:00 2001
From: adrinjalali <adrin.jalali@gmail.com>
Date: Tue, 5 Dec 2023 12:18:55 +0100
Subject: [PATCH] TST expect warning in API call

---
 skops/hub_utils/tests/test_hf_hub.py | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/skops/hub_utils/tests/test_hf_hub.py b/skops/hub_utils/tests/test_hf_hub.py
index e4f8ddbb..2326c80a 100644
--- a/skops/hub_utils/tests/test_hf_hub.py
+++ b/skops/hub_utils/tests/test_hf_hub.py
@@ -433,15 +433,16 @@ def test_inference(
     user = client.whoami(token=HF_HUB_TOKEN)["name"]
     repo_id = f"{user}/test-{uuid4()}"
 
-    push(
-        repo_id=repo_id,
-        source=destination_path,
-        token=HF_HUB_TOKEN,
-        commit_message="test message",
-        create_remote=True,
-        # api-inference doesn't support private repos for community projects.
-        private=False,
-    )
+    with pytest.warns(FutureWarning, match="Creating repos on hf.co is subject"):
+        push(
+            repo_id=repo_id,
+            source=destination_path,
+            token=HF_HUB_TOKEN,
+            commit_message="test message",
+            create_remote=True,
+            # api-inference doesn't support private repos for community projects.
+            private=False,
+        )
 
     X_test = data.data.head(5)
     y_pred = model.predict(X_test)