Adding wrapper for calculate_moment.

src/qp/metrics/base_metric_classes.py

@@ -102,23 +102,41 @@ class SingleEnsembleMetric(BaseMetric):
     metric_input_type = MetricInputType.single_ensemble
     metric_output_type = MetricOuputType.one_value_per_distribution
 
+    def initialize(self):
+        raise NotImplementedError()
+
+    def evaluate(self, estimate):
+        raise NotImplementedError()
+
+    def finalize(self):
+        raise NotImplementedError()
+
 
 class DistToDistMetric(BaseMetric):
 
     metric_input_type = MetricInputType.dist_to_dist
 
+    def initialize(self):
+        raise NotImplementedError()
+
+    def evaluate(self, estimate, reference):
+        raise NotImplementedError()
+
+    def finalize(self):
+        raise NotImplementedError()
+
 
 class DistToPointMetric(BaseMetric):
 
     metric_input_type = MetricInputType.dist_to_point
 
-    def initialize(self, **kwargs):
+    def initialize(self):
         raise NotImplementedError()
 
-    def evaluate(self, estimate, reference, **kwargs):
+    def evaluate(self, estimate, reference):
         raise NotImplementedError()
 
-    def finalize(self, **kwargs):
+    def finalize(self):
         raise NotImplementedError()
 
 

src/qp/metrics/concrete_metric_classes.py

@@ -1,5 +1,27 @@
-from qp.metrics.base_metric_classes import MetricOuputType, DistToDistMetric
-from qp.metrics.metrics import calculate_kld
+from qp.metrics.base_metric_classes import MetricOuputType, DistToDistMetric, SingleEnsembleMetric
+from qp.metrics.metrics import calculate_kld, calculate_moment
+
+class MomentMetric(SingleEnsembleMetric):
+    """Class wrapper around the `calculate_moment` function.
+    """
+
+    metric_name = "moment"
+
+    def __init__(self, moment_order:int=1, limits:tuple=(0.0, 3.0), dx:float=0.01) -> None:
+        super().__init__()
+        self._moment_order = moment_order
+        self._limits = limits
+        self._dx = dx
+
+    def initialize(self) -> None:
+        pass
+
+    def evalulate(self, estimate) -> list:
+        return calculate_moment(estimate, self._moment_order, self._limits, self._dx)
+
+    def finalize(self) -> None:
+        pass
+
 
 class KLDMetric(DistToDistMetric):
     """Class wrapper around the KLD metric
@@ -16,7 +38,7 @@ def __init__(self, limits:tuple=(0.0, 3.0), dx:float=0.01) -> None:
     def initialize(self) -> None:
         pass
 
-    def evaluate(self, estimate, reference) -> None:
+    def evaluate(self, estimate, reference) -> list:
         return calculate_kld(estimate, reference, self._limits, self._dx)
 
     def finalize(self) -> None: