From d784c49566e5a7705539fb4ceb0461bad8f41361 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 11 Sep 2023 22:10:45 +0200 Subject: Update imports --- text_recognizer/data/base_data_module.py | 1 - text_recognizer/data/base_dataset.py | 2 +- text_recognizer/data/transforms/pad.py | 1 - 3 files changed, 1 insertion(+), 3 deletions(-) (limited to 'text_recognizer/data') diff --git a/text_recognizer/data/base_data_module.py b/text_recognizer/data/base_data_module.py index 3cb0aa2..6efa4fb 100644 --- a/text_recognizer/data/base_data_module.py +++ b/text_recognizer/data/base_data_module.py @@ -1,5 +1,4 @@ """Base lightning DataModule class.""" -from pathlib import Path from typing import Callable, Dict, Optional, Tuple, TypeVar import pytorch_lightning as L diff --git a/text_recognizer/data/base_dataset.py b/text_recognizer/data/base_dataset.py index b840bc8..12f524c 100644 --- a/text_recognizer/data/base_dataset.py +++ b/text_recognizer/data/base_dataset.py @@ -1,5 +1,5 @@ """Base PyTorch Dataset class.""" -from typing import Callable, Dict, Optional, Sequence, Tuple, Union +from typing import Callable, Dict, Sequence, Tuple, Union import torch from torch import Tensor diff --git a/text_recognizer/data/transforms/pad.py b/text_recognizer/data/transforms/pad.py index df1d83f..baf637a 100644 --- a/text_recognizer/data/transforms/pad.py +++ b/text_recognizer/data/transforms/pad.py @@ -1,7 +1,6 @@ """Pad targets to equal length.""" import torch -import torch.functional as F from torch import Tensor -- cgit v1.2.3-70-g09d2