Skip to content

Commit d2bf709

Browse files
committed
Add Hugging Face model support to Shape Recommender
1 parent 4c2c6a2 commit d2bf709

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/unitary/with_extras/aqua/test_recommend.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -454,7 +454,7 @@ def test_shape_report_pareto_front(self):
454454
assert c and d in pf
455455
assert a and b not in pf
456456
assert len(pf) == 2
457-
# --- Tests for HuggingFaceModelFetcher ---
457+
458458
class TestHuggingFaceModelFetcher:
459459
@pytest.mark.parametrize("model_id, expected", [
460460
("meta-llama/Llama-2-7b-hf", True),
@@ -488,20 +488,17 @@ def test_fetch_config_only_not_found(self, mock_get):
488488
def test_get_hf_token(self):
489489
assert HuggingFaceModelFetcher.get_hf_token() == "test_token_123"
490490

491-
# Add this method inside the TestHuggingFaceModelFetcher class
492491

493492
@pytest.mark.network
494493
def test_fetch_config_only_real_call_success(self):
495494
"""
496495
Tests a real network call to fetch a public model's configuration.
497496
This test requires an internet connection.
498497
"""
499-
# Use a well-known, small, public model to minimize test flakiness
500498
model_id = "distilbert-base-uncased"
501499

502500
try:
503501
config = HuggingFaceModelFetcher.fetch_config_only(model_id)
504-
# Assert that we got a dictionary with expected keys
505502
assert isinstance(config, dict)
506503
assert "model_type" in config
507504
assert "dim" in config

0 commit comments

Comments
 (0)