Skip to content

Commit 9000c34

Browse files
minor changes
1 parent a677db7 commit 9000c34

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

ads/aqua/model/model.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
)
3636
from ads.aqua.model.constants import *
3737
from ads.aqua.model.entities import *
38+
from ads.common.auth import default_signer
3839
from ads.common.oci_resource import SEARCH_TYPE, OCIResource
3940
from ads.common.utils import get_console_link, is_path_exists
4041
from ads.config import (
@@ -679,8 +680,8 @@ def register(
679680
finetuning_container (str): selects service defaults
680681
681682
Returns:
682-
DataScienceModel:
683-
The registered model as a DataScienceModel object.
683+
AquaModel:
684+
The registered model as a AquaModel object.
684685
"""
685686
verified_model_details: DataScienceModel = None
686687

@@ -838,7 +839,7 @@ def load_license(self, model_id: str) -> AquaModelLicense:
838839
content = str(
839840
read_file(
840841
file_path=f"{os.path.dirname(artifact_path)}/{LICENSE_TXT}",
841-
auth=self._auth,
842+
auth=default_signer(),
842843
)
843844
)
844845

0 commit comments

Comments
 (0)