diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-09-11 22:10:45 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-09-11 22:10:45 +0200 |
| commit | d784c49566e5a7705539fb4ceb0461bad8f41361 (patch) | |
| tree | a925281a3b335a9df3c0500d61f6571616275e26 /text_recognizer/data | |
| parent | 43b10ea57f40983ec53c17a05f5b9fd2501f4eea (diff) | |
Update imports
Diffstat (limited to 'text_recognizer/data')
| -rw-r--r-- | text_recognizer/data/base_data_module.py | 1 | ||||
| -rw-r--r-- | text_recognizer/data/base_dataset.py | 2 | ||||
| -rw-r--r-- | text_recognizer/data/transforms/pad.py | 1 |
3 files changed, 1 insertions, 3 deletions
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 |