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

Fix typing and minor issues

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-07 14:12:44 +02:00
parent 2db2253578
commit 72ab627a34
54 changed files with 635 additions and 589 deletions
Download patch file Download diff file Expand all files Collapse all files

6
great_ai/models/use_model.py
View file

@ -21,7 +21,7 @@ from ..helper.assert_function_is_not_finalised import assert_function_is_not_fin
from ..tracing.tracing_context import TracingContext
from ..views import Model
F = TypeVar("F", bound=Callable[..., Any])
F = TypeVar("F", bound=Callable)
def use_model(
@ -70,4 +70,6 @@ def _load_model(key: str, version: Optional[int] = None) -> Tuple[Any, int]:
return path, file.version
with file as f:
return load(f), file.version
loaded = load(f)
return loaded, file.version