added baseline name
This commit is contained in:
parent
8096c0842e
commit
3180622787
|
@ -28,6 +28,7 @@ def baseline(func):
|
||||||
def wrapper(c_model, validation, protocol):
|
def wrapper(c_model, validation, protocol):
|
||||||
return func(c_model, validation, protocol)
|
return func(c_model, validation, protocol)
|
||||||
|
|
||||||
|
wrapper.name = func.__name__
|
||||||
_baselines[func.__name__] = wrapper
|
_baselines[func.__name__] = wrapper
|
||||||
|
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
Loading…
Reference in New Issue