andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Simplify API

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-01 19:50:43 +02:00
commit b56273e267
1 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/great_ai/great_ai/models/use_model.py
View file

@ -14,7 +14,6 @@ def use_model(
key: str,
*,
version: Union[int, Literal["latest"]],
return_path: bool = False,
model_kwarg_name: str = "model",
) -> Callable[[F], F]:
assert (
@ -24,7 +23,6 @@ def use_model(
model, actual_version = load_model(
key=key,
version=None if version == "latest" else version,
return_path=return_path,
)
def decorator(func: F) -> F: