Formatting?

Author: Seilmast

utils/load_metric.py

@@ -90,6 +90,6 @@ def __getmetrics__(self, str_prefix: str = None):
                 return_metrics[key] = self.metrics[key].__returnmetric__()
         return return_metrics
 
-    def reset(self):
+    def __resetmetrics__(self):
         for key in self.metrics:
-            self.metrics[key].reset()
+            self.metrics[key].__reset__()

utils/metrics/EntropyPred.py

@@ -50,15 +50,15 @@ def __call__(self, y_true, y_logits):
 
         return entropy_values
 
-    def return_value(self):
+    def __returnmetric__(self):
         if self.averages == "mean":
             self.stored_entropy_values = th.mean(self.stored_entropy_values)
         elif self.averages == "sum":
             self.stored_entropy_values = th.sum(self.stored_entropy_values)
         elif self.averages == "none":
             return self.stored_entropy_values
 
-    def reset(self):
+    def __reset__(self):
         self.stored_entropy_values = []