From eae9024fcb2518087b9b9d1b11fee4feab84a15f Mon Sep 17 00:00:00 2001 From: Prasad Jivane Date: Sun, 2 Apr 2023 00:42:12 +0530 Subject: [PATCH] Update custom_training_loop.py Removed unused import --- core/custom_training_loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/custom_training_loop.py b/core/custom_training_loop.py index 0241145..a8f13fb 100644 --- a/core/custom_training_loop.py +++ b/core/custom_training_loop.py @@ -10,7 +10,7 @@ Features (go/100_enablements): """ import datetime import os -from typing import Callable, Dict, Iterable, List, Mapping, Optional +from typing import Callable, Dict, Iterable, List, Optional from tml.common import log_weights