fix: add model argument to metrics (#118)

flamapy · Jan 12, 2024 · 4ee9ae3 · 4ee9ae3
1 parent 8c485b1
commit 4ee9ae3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/flamapy/core/operations/metrics_operation.py b/flamapy/core/operations/metrics_operation.py
@@ -22,7 +22,10 @@ def __init__(self) -> None:
         self.model: Optional[VariabilityModel] = None
 
     @abstractmethod
-    def calculate_metamodel_metrics(self) -> List[Dict[str, Any]]:
+    def calculate_metamodel_metrics(
+        self,
+        model: VariabilityModel,
+    ) -> List[Dict[str, Any]]:
         """Return a list of metrics for each metamodel"""  
 
     def only_these_metrics(self, filter: List[str]) -> None:
@@ -99,4 +102,4 @@ def construct_result(
 
     @staticmethod
     def only_these_metrics(self, filter: List[str]) -> None:
-        self.filter = filter 
+        self.filter = filter