diff --git a/core/metrics.py b/core/metrics.py
index 2384e4d..cdbc50c 100644
--- a/core/metrics.py
+++ b/core/metrics.py
@@ -5,7 +5,11 @@ We assume multi task models will output [task_idx, ...] predictions
 """
 from typing import Any, Dict
 
-from tml.core.metric_mixin import MetricMixin, StratifyMixin, TaskMixin
+from tml.core.metric_mixin import (
+  MetricMixin,
+  StratifyMixin,
+  TaskMixin
+)
 
 import torch
 import torchmetrics as tm