From c697f18583ee1d8e2e65f040b600c621416f1dbc Mon Sep 17 00:00:00 2001 From: Pavel Bezglasny Date: Sat, 11 Dec 2021 00:05:47 +0300 Subject: [PATCH] Change return type of method get_latest_versions to List[ModelVersion] Signed-off-by: Pavel Bezglasny --- mlflow/tracking/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlflow/tracking/client.py b/mlflow/tracking/client.py index f1e06dae2609e..59079a9a59326 100644 --- a/mlflow/tracking/client.py +++ b/mlflow/tracking/client.py @@ -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.