From 415863550d563a5337636c6a0424e9fbfccc17ef Mon Sep 17 00:00:00 2001 From: Siddhi Kiran Bajracharyya Date: Mon, 3 Apr 2023 17:59:30 -0500 Subject: [PATCH] Update metrics.py multiline import --- core/metrics.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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