From 415863550d563a5337636c6a0424e9fbfccc17ef Mon Sep 17 00:00:00 2001 From: Siddhi Kiran Bajracharyya Date: Mon, 3 Apr 2023 17:59:30 -0500 Subject: [PATCH 1/2] 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 @@ """ 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 From 7d9ac9a6439b43a0e6b1315a73e2332144125671 Mon Sep 17 00:00:00 2001 From: Siddhi Kiran Bajracharyya Date: Mon, 3 Apr 2023 18:00:58 -0500 Subject: [PATCH 2/2] Update custom_training_loop.py formatted imports --- core/custom_training_loop.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/core/custom_training_loop.py b/core/custom_training_loop.py index 0241145..57aba03 100644 --- a/core/custom_training_loop.py +++ b/core/custom_training_loop.py @@ -8,14 +8,22 @@ - go/dataset-service 0-copy integration """ -import datetime import os -from typing import Callable, Dict, Iterable, List, Mapping, Optional +import datetime +from typing import ( + Callable, + Dict, + Iterable, + List, + Mapping, + Optional +) + from tml.common import log_weights -import tml.common.checkpointing.snapshot as snapshot_lib from tml.core.losses import get_global_loss_detached +import tml.common.checkpointing.snapshot as snapshot_lib from tml.ml_logging.torch_logging import logging # type: ignore[attr-defined] from tml.core.train_pipeline import TrainPipelineSparseDist