Skip to content

Commit 367ce70

Browse files
committed
removing real network call
1 parent 76288dd commit 367ce70

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

tests/unitary/with_extras/aqua/test_recommend.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -489,21 +489,21 @@ def test_get_hf_token(self):
489489
assert HuggingFaceModelFetcher.get_hf_token() == "test_token_123"
490490

491491

492-
@pytest.mark.network
493-
def test_fetch_config_only_real_call_success(self):
494-
"""
495-
Tests a real network call to fetch a public model's configuration.
496-
This test requires an internet connection.
497-
"""
498-
model_id = "distilbert-base-uncased"
492+
# @pytest.mark.network
493+
# def test_fetch_config_only_real_call_success(self):
494+
# """
495+
# Tests a real network call to fetch a public model's configuration.
496+
# This test requires an internet connection.
497+
# """
498+
# model_id = "distilbert-base-uncased"
499499

500-
try:
501-
config = HuggingFaceModelFetcher.fetch_config_only(model_id)
502-
assert isinstance(config, dict)
503-
assert "model_type" in config
504-
assert "dim" in config
505-
except AquaValueError as e:
506-
pytest.fail(f"Real network call to Hugging Face failed: {e}")
500+
# try:
501+
# config = HuggingFaceModelFetcher.fetch_config_only(model_id)
502+
# assert isinstance(config, dict)
503+
# assert "model_type" in config
504+
# assert "dim" in config
505+
# except AquaValueError as e:
506+
# pytest.fail(f"Real network call to Hugging Face failed: {e}")
507507

508508

509509
@patch('ads.aqua.shaperecommend.recommend.OCIDataScienceModelDeployment.shapes')

0 commit comments

Comments
 (0)