Update custom_training_loop.py

Removed unused import
This commit is contained in:
Prasad Jivane 2023-04-02 00:42:12 +05:30 committed by GitHub
parent 78c3235eee
commit eae9024fcb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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