From 3180622787eb4512d9cd6532724311e4b0db9aa1 Mon Sep 17 00:00:00 2001 From: Lorenzo Volpi Date: Sun, 26 Nov 2023 16:33:56 +0100 Subject: [PATCH] added baseline name --- quacc/evaluation/baseline.py | 1 + 1 file changed, 1 insertion(+) 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