Skip to content

Commit 0eecc43

Browse files
authored
fix compatibility with SDK, _parse_org_teamspace_model_version (#100)
1 parent 52a5b8b commit 0eecc43

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
lightning-sdk >=0.2.9
1+
lightning-sdk >=0.2.10
22
lightning-utilities

src/litmodels/io/cloud.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from typing import TYPE_CHECKING, Dict, List, Optional, Union
77

88
from lightning_sdk.lightning_cloud.env import LIGHTNING_CLOUD_URL
9-
from lightning_sdk.models import _extend_model_name_with_teamspace, _parse_model_name_and_version
9+
from lightning_sdk.models import _extend_model_name_with_teamspace, _parse_org_teamspace_model_version
1010
from lightning_sdk.models import download_model as sdk_download_model
1111
from lightning_sdk.models import upload_model as sdk_upload_model
1212

@@ -31,7 +31,7 @@ def _print_model_link(name: str, verbose: Union[bool, int]) -> None:
3131
- If set to 2, the link will be printed every time.
3232
"""
3333
name = _extend_model_name_with_teamspace(name)
34-
org_name, teamspace_name, model_name, _ = _parse_model_name_and_version(name)
34+
org_name, teamspace_name, model_name, _ = _parse_org_teamspace_model_version(name)
3535

3636
url = f"{LIGHTNING_CLOUD_URL}/{org_name}/{teamspace_name}/models/{model_name}"
3737
msg = f"Model uploaded successfully. Link to the model: '{url}'"

0 commit comments

Comments
 (0)