ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-04-25 16:39:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add request model detection in endpoint decorator

Browse Source
This commit is contained in:
allegroai 2021-01-05 18:47:01 +02:00
parent f879f6924f
commit f9f2f0ccf0
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apiserver/service_repo/__init__.py
View File

@ -28,7 +28,13 @@ def endpoint(
# Backwards compatibility: support endpoints with both old-style signature (call) and new-style signature
# (call, company, request_model)
func = f
if len(signature(f).parameters) == 1:
sig = signature(f)
nonlocal request_data_model
if len(sig.parameters) == 3 and request_data_model is None:
last_arg = list(sig.parameters.items())[-1][1]
if last_arg.annotation != last_arg.empty:
request_data_model = last_arg.annotation
elif len(sig.parameters) == 1:
# old-style
def adapter(call, *_, **__):
return f(call)