diff --git a/requirements.txt b/requirements.txt index 3ab0459..e0a1323 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ -lightning-sdk >=0.2.9 +lightning-sdk >=0.2.10 lightning-utilities diff --git a/src/litmodels/io/cloud.py b/src/litmodels/io/cloud.py index 8af5672..8cff2a1 100644 --- a/src/litmodels/io/cloud.py +++ b/src/litmodels/io/cloud.py @@ -6,7 +6,7 @@ from typing import TYPE_CHECKING, Dict, List, Optional, Union from lightning_sdk.lightning_cloud.env import LIGHTNING_CLOUD_URL -from lightning_sdk.models import _extend_model_name_with_teamspace, _parse_model_name_and_version +from lightning_sdk.models import _extend_model_name_with_teamspace, _parse_org_teamspace_model_version from lightning_sdk.models import download_model as sdk_download_model from lightning_sdk.models import upload_model as sdk_upload_model @@ -31,7 +31,7 @@ def _print_model_link(name: str, verbose: Union[bool, int]) -> None: - If set to 2, the link will be printed every time. """ name = _extend_model_name_with_teamspace(name) - org_name, teamspace_name, model_name, _ = _parse_model_name_and_version(name) + org_name, teamspace_name, model_name, _ = _parse_org_teamspace_model_version(name) url = f"{LIGHTNING_CLOUD_URL}/{org_name}/{teamspace_name}/models/{model_name}" msg = f"Model uploaded successfully. Link to the model: '{url}'"