Adopt type hinting generics in standard collections (PEP 585)

1 files changed, 2 insertions, 2 deletions

diff --git a/utils/sampler.py b/utils/sampler.py
index 8dec846..0a177d1 100644
--- a/utils/sampler.py
+++ b/utils/sampler.py
@@ -1,5 +1,5 @@
 import random
-from typing import Iterator, Tuple
+from collections.abc import Iterator
 
 import numpy as np
 from torch.utils import data
@@ -11,7 +11,7 @@ class TripletSampler(data.Sampler):
     def __init__(
             self,
             data_source: CASIAB,
-            batch_size: Tuple[int, int]
+            batch_size: tuple[int, int]
     ):
         super().__init__(data_source)
         self.metadata_labels = data_source.metadata['labels']