Skip to content

Commit 43be7fc

Browse files
authoredAug 3, 2023
[HWORKS-682] Include return type and description for model.download() (logicalclocks#197)
1 parent 66ee2bc commit 43be7fc

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed
 

‎python/hsml/model.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -90,14 +90,18 @@ def save(self, model_path, await_registration=480):
9090
await_registration: Awaiting time for the model to be registered in Hopsworks.
9191
9292
# Returns
93-
`Model`. The model metadata object.
93+
`Model`: The model metadata object.
9494
"""
9595
return self._model_engine.save(
9696
self, model_path, await_registration=await_registration
9797
)
9898

9999
def download(self):
100-
"""Download the model files to a local folder."""
100+
"""Download the model files.
101+
102+
# Returns
103+
`str`: Absolute path to local folder containing the model files.
104+
"""
101105
return self._model_engine.download(self)
102106

103107
def delete(self):
@@ -154,7 +158,7 @@ def deploy(
154158
transformer: Transformer to be deployed together with the predictor.
155159
156160
# Returns
157-
`Deployment`. The deployment metadata object of a new or existing deployment.
161+
`Deployment`: The deployment metadata object of a new or existing deployment.
158162
"""
159163

160164
if name is None:

0 commit comments

Comments
 (0)