andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Add more useful return value

Browse source
This commit is contained in:
Andras Schmelczer 2022-05-27 11:13:02 +02:00
parent 2c516cf861
commit c0c88d6059
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
great_ai/src/great_ai/great_ai/models/save_model.py
View file

@ -10,7 +10,7 @@ from ..context import get_context
def save_model( def save_model(
model: Union[Path, str, object], key: str, keep_last_n: Optional[int] = None 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 get_context() # will setup LargeFile if there was no config set
file = LargeFile(name=key, mode="wb", keep_last_n=keep_last_n) 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}") get_context().logger.info(f"Model {key} uploaded with version {file.version}")
return file.version return f"{key}:{file.version}"