Skip to content

Commit

Permalink
Change return type of method get_latest_versions to List[ModelVersion] (
Browse files Browse the repository at this point in the history
#5156)

Signed-off-by: Pavel Bezglasny <pavelb23@gmail.com>
  • Loading branch information
pbezglasny committed Dec 11, 2021
1 parent c630d6d commit 0da7c6c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mlflow/tracking/client.py
Expand Up @@ -1980,7 +1980,7 @@ def print_model_info(rm):
"""
return self._get_registry_client().get_registered_model(name)

def get_latest_versions(self, name: str, stages: List[str] = None) -> ModelVersion:
def get_latest_versions(self, name: str, stages: List[str] = None) -> List[ModelVersion]:
"""
Latest version models for each requests stage. If no ``stages`` provided, returns the
latest version for each stage.
Expand Down

0 comments on commit 0da7c6c

Please sign in to comment.