Fixes a bug in registering HF models in AQUA UI (#1201)

Author: mrDzurb

ads/aqua/extension/model_handler.py

@@ -241,7 +241,7 @@ def _find_matching_aqua_model(model_id: str) -> Optional[AquaModelSummary]:
         aqua_model_app = AquaModelApp()
         model_ocid = aqua_model_app._find_matching_aqua_model(model_id=model_id_lower)
         if model_ocid:
-            return aqua_model_app.get(model_ocid, load_model_card=False)
+            return aqua_model_app.get(model_ocid)
 
         return None
 

ads/aqua/model/model.py

@@ -491,8 +491,6 @@ def get(self, model_id: str) -> "AquaModel":
         ----------
         model_id: str
             The model OCID.
-        load_model_card: (bool, optional). Defaults to `True`.
-            Whether to load model card from artifacts or not.
 
         Returns
         -------

tests/unitary/with_extras/aqua/test_model_handler.py

@@ -14,10 +14,7 @@
 
 from ads.aqua.common.errors import AquaRuntimeError
 from ads.aqua.common.utils import get_hf_model_info
-from ads.aqua.constants import (
-    AQUA_TROUBLESHOOTING_LINK,
-    STATUS_CODE_MESSAGES,
-)
+from ads.aqua.constants import AQUA_TROUBLESHOOTING_LINK, STATUS_CODE_MESSAGES
 from ads.aqua.extension.errors import ReplyDetails
 from ads.aqua.extension.model_handler import (
     AquaHuggingFaceHandler,
@@ -27,7 +24,7 @@
 )
 from ads.aqua.model import AquaModelApp
 from ads.aqua.model.entities import AquaModel, AquaModelSummary, HFModelSummary
-from ads.config import USER, SERVICE
+from ads.config import SERVICE, USER
 
 
 class ModelHandlerTestCase(TestCase):
@@ -346,9 +343,7 @@ def test_find_matching_aqua_model(
             aqua_model_mock_list.assert_called_once()
 
             if expected_aqua_model_name:
-                mock_get_model.assert_called_with(
-                    expected_aqua_model_id, load_model_card=False
-                )
+                mock_get_model.assert_called_with(expected_aqua_model_id)
             else:
                 assert test_result == None