v0.1.8

robogauge/tasks/gauge/metrics/base_metric.py

@@ -1,6 +1,7 @@
 from robogauge.utils.logger import logger
 from robogauge.tasks.robots import RobotConfig
 from robogauge.tasks.simulator.sim_data import SimData
+from robogauge.tasks.gauge.goals.base_goal import GoalData
 
 class BaseMetric:
     """ Base class for all metric functions. """
@@ -9,7 +10,7 @@ class BaseMetric:
     def __init__(self, robot_cfg: RobotConfig, **kwargs):
         self.robot_cfg = robot_cfg
 
-    def __call__(self, sim_data: SimData) -> float:
+    def __call__(self, sim_data: SimData, goal_data: GoalData) -> float:
         value = 0.0
         logger.log(value, self.name, step=sim_data.n_step)
         return value