From c0c88d6059d8bcffb8d5dfebb8caaeec639e0e6e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A1s=20Schmelczer?= <andras@schmelczer.dev>
Date: Fri, 27 May 2022 11:13:02 +0200
Subject: [PATCH] Add more useful return value

---
 great_ai/src/great_ai/great_ai/models/save_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/great_ai/src/great_ai/great_ai/models/save_model.py b/great_ai/src/great_ai/great_ai/models/save_model.py
index 338fcd2..6aef357 100644
--- a/great_ai/src/great_ai/great_ai/models/save_model.py
+++ b/great_ai/src/great_ai/great_ai/models/save_model.py
@@ -10,7 +10,7 @@ from ..context import get_context
 
 def save_model(
     model: Union[Path, str, object], key: str, keep_last_n: Optional[int] = None
-) -> int:
+) -> str:
     get_context()  # will setup LargeFile if there was no config set
 
     file = LargeFile(name=key, mode="wb", keep_last_n=keep_last_n)
@@ -23,4 +23,4 @@ def save_model(
 
     get_context().logger.info(f"Model {key} uploaded with version {file.version}")
 
-    return file.version
+    return f"{key}:{file.version}"