get_fitted: also provide information on static (sub)models

1 files changed, 7 insertions, 0 deletions

diff --git a/lib/functions.py b/lib/functions.py
index 067514f..0f0bf47 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -157,6 +157,13 @@ class ModelInfo:
         pass
 
 
+class StaticInfo:
+    def __init__(self, data):
+        self.mean = np.mean(data)
+        self.median = np.median(data)
+        self.std = np.std(data)
+
+
 class AnalyticInfo(ModelInfo):
     def __init__(self, fit_result, function):
         self.fit_result = fit_result