fix: recover eligible model GPU routes
This commit is contained in:
@@ -93,6 +93,31 @@ class ModelScopeDiscoveryTests(unittest.TestCase):
|
||||
self.assertEqual([], models)
|
||||
self.assertEqual(50, http_client.queries[0]["page_size"])
|
||||
|
||||
def test_popular_sort_uses_separate_page_cache(self) -> None:
|
||||
http_client = RecordingHttpClient()
|
||||
discovery = HuggingFaceDiscovery(http_client=http_client) # type: ignore[arg-type]
|
||||
|
||||
discovery.list_recent_models(pipeline_tags=["text-generation"], limit=50, min_downloads=0)
|
||||
discovery.list_recent_models(
|
||||
pipeline_tags=["text-generation"], limit=50, min_downloads=0, sort="downloads"
|
||||
)
|
||||
self.assertEqual(["last_modified", "downloads"], [item["sort"] for item in http_client.queries])
|
||||
|
||||
def test_popular_scan_rotates_bounded_page_windows(self) -> None:
|
||||
http_client = RecordingHttpClient()
|
||||
discovery = HuggingFaceDiscovery(http_client=http_client) # type: ignore[arg-type]
|
||||
|
||||
for _ in range(2):
|
||||
discovery.list_recent_models(
|
||||
pipeline_tags=["text-generation"],
|
||||
limit=200,
|
||||
min_downloads=50,
|
||||
sort="downloads",
|
||||
rotate_pages=True,
|
||||
)
|
||||
|
||||
self.assertEqual([1, 5], [item["page_number"] for item in http_client.queries])
|
||||
|
||||
def test_rate_limited_page_keeps_previous_pages_and_resumes_from_cache(self) -> None:
|
||||
http_client = RateLimitedPageClient()
|
||||
discovery = HuggingFaceDiscovery(
|
||||
|
||||
Reference in New Issue
Block a user