Skip to content

Commit 556bcec

Browse files
committed
Handle model publish link references without cls
1 parent 5d098e5 commit 556bcec

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/sasctl/_services/model_publish.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,10 +112,12 @@ def publish_model(cls, model, destination, name=None, code=None, notes=None):
112112
-------
113113
114114
"""
115-
code_types = {"ds2package": "ds2", "datastep": "datastep", "": ""}
115+
code_types = {"ds2package": "ds2", "datastep": "datastep", "python": "python"}
116116

117117
model = cls._model_repository.get_model(model)
118-
model_uri = cls._model_repository.get_model_link(model, "cls")
118+
model_uri = cls._model_repository.get_model_link(model, "self", True)
119+
if not model_uri:
120+
model_uri = cls._model_repository.get_model_link(model, "cls")
119121

120122
# Get score code from registry if no code specified
121123
if code is None:

0 commit comments

Comments
 (0)