Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
lightning-sdk >=0.2.9
lightning-sdk >=0.2.10
lightning-utilities
4 changes: 2 additions & 2 deletions src/litmodels/io/cloud.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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}'"
Expand Down
Loading