File tree Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Original file line number Diff line number Diff 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+
458458class 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
You can’t perform that action at this time.
0 commit comments