diff --git a/quacc/evaluation/baseline.py b/quacc/evaluation/baseline.py index c313bed..eca0b81 100644 --- a/quacc/evaluation/baseline.py +++ b/quacc/evaluation/baseline.py @@ -28,6 +28,7 @@ def baseline(func): def wrapper(c_model, validation, protocol): return func(c_model, validation, protocol) + wrapper.name = func.__name__ _baselines[func.__name__] = wrapper return wrapper