def test_model_list(self): _api = HfApi() models = _api.model_list() self.assertGreater(len(models), 100) self.assertIsInstance(models[0], ModelInfo)
def test_staging_model_list(self): _api = HfApi(endpoint=ENDPOINT_STAGING) _ = _api.model_list()