1 files changed, 2 insertions, 3 deletions
diff --git a/utils/sampler.py b/utils/sampler.py
index 0c9872c..581d7a2 100644
--- a/utils/sampler.py
+++ b/utils/sampler.py
@@ -1,6 +1,5 @@
 import random
-from collections.abc import Iterator
-from typing import Union
+from typing import Union, Tuple, Iterator
 
 import numpy as np
 from torch.utils import data
@@ -12,7 +11,7 @@ class TripletSampler(data.Sampler):
     def __init__(
             self,
             data_source: Union[CASIAB],
-            batch_size: tuple[int, int]
+            batch_size: Tuple[int, int]
     ):
         super().__init__(data_source)
         self.metadata_labels = data_source.metadata['labels']