diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-27 11:15:51 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-27 11:15:51 +0800 |
commit | 81b12f9357ca737599fbc11c732020666ad0873f (patch) | |
tree | 21beabf4ad44a0098478e143d8032bb42e5e92c0 /utils/configuration.py | |
parent | 31b20fbff0786c998c54b8585de759d02f41eda7 (diff) |
Adopt type hinting generics in standard collections (PEP 585)
Diffstat (limited to 'utils/configuration.py')
-rw-r--r-- | utils/configuration.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/utils/configuration.py b/utils/configuration.py index 32b9bec..84bd064 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -1,4 +1,4 @@ -from typing import TypedDict, Tuple +from typing import TypedDict import torch @@ -16,12 +16,12 @@ class DatasetConfiguration(TypedDict): num_sampled_frames: int discard_threshold: int num_input_channels: int - frame_size: Tuple[int, int] + frame_size: tuple[int, int] cache_on: bool class DataloaderConfiguration(TypedDict): - batch_size: Tuple[int, int] + batch_size: tuple[int, int] num_workers: int pin_memory: bool @@ -29,7 +29,7 @@ class DataloaderConfiguration(TypedDict): class HyperparameterConfiguration(TypedDict): hidden_dim: int lr: int - betas: Tuple[float, float] + betas: tuple[float, float] hard_or_all: str margin: float |