Skip to content

Commit 52077ce

Browse files
Reverting changes
1 parent 2174b36 commit 52077ce

File tree

3 files changed

+1
-36
lines changed

3 files changed

+1
-36
lines changed

Makefile

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,4 @@ clean:
1010
@find ./ -name '*.pyc' -exec rm -f {} \;
1111
@find ./ -name 'Thumbs.db' -exec rm -f {} \;
1212
@find ./ -name '*~' -exec rm -f {} \;
13-
@find ./ -name '.DS_Store' -exec rm -f {} \;
14-
test:
15-
pip install -e .
16-
jupyter server extension enable --py ads.aqua.extension
17-
jupyter lab --NotebookApp.disable_check_xsrf=True --no-browser
13+
@find ./ -name '.DS_Store' -exec rm -f {} \;

ads/aqua/common/utils.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -57,15 +57,13 @@
5757
VLLM_INFERENCE_RESTRICTED_PARAMS,
5858
)
5959
from ads.aqua.data import AquaResourceIdentifier
60-
from ads.aqua.model.constants import ModelTask
6160
from ads.common.auth import AuthState, default_signer
6261
from ads.common.extended_enum import ExtendedEnumMeta
6362
from ads.common.object_storage_details import ObjectStorageDetails
6463
from ads.common.oci_resource import SEARCH_TYPE, OCIResource
6564
from ads.common.utils import copy_file, get_console_link, upload_to_os
6665
from ads.config import AQUA_SERVICE_MODELS_BUCKET, CONDA_BUCKET_NS, TENANCY_OCID
6766
from ads.model import DataScienceModel, ModelVersionSet
68-
from tests.unitary.with_extras.model.score import model_name
6967

7068
logger = logging.getLogger("ads.aqua")
7169

@@ -1065,11 +1063,5 @@ def get_hf_model_info(repo_id: str) -> ModelInfo:
10651063
except HfHubHTTPError as err:
10661064
raise format_hf_custom_error_message(err) from err
10671065

1068-
def list_hf_models(query:str) -> List[str]:
1069-
try:
1070-
models= HfApi().list_models(model_name=query,task=ModelTask.TEXT_GENERATION)
1071-
return [model.id for model in models if model.disabled is None]
1072-
except HfHubHTTPError as err:
1073-
raise format_hf_custom_error_message(err) from err
10741066

10751067

ads/aqua/extension/model_handler.py

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -177,29 +177,6 @@ def _find_matching_aqua_model(model_id: str) -> Optional[AquaModelSummary]:
177177

178178
return None
179179

180-
@handle_exceptions
181-
def get(self):
182-
"""
183-
Finds a list of matching models from hugging face based on query string provided from users.
184-
185-
Parameters
186-
----------
187-
query (str): The Hugging Face model name to search for.
188-
189-
Returns
190-
-------
191-
List[AquaModelSummary]
192-
Returns the matching AquaModelSummary object if found, else None.
193-
"""
194-
195-
query=self.get_argument("query",default=None)
196-
if not query:
197-
raise HTTPError(400,Errors.MISSING_REQUIRED_PARAMETER.format("query"))
198-
models=list_hf_models(query)
199-
return self.finish({"models":models})
200-
201-
202-
203180
@handle_exceptions
204181
def post(self, *args, **kwargs):
205182
"""Handles post request for the HF Models APIs

0 commit comments

Comments
 (0)