1 files changed, 2 insertions, 2 deletions
diff --git a/utils/sampler.py b/utils/sampler.py
index cdf1984..734acf9 100644
--- a/utils/sampler.py
+++ b/utils/sampler.py
@@ -1,6 +1,6 @@
 import random
 from collections.abc import Iterator
-from typing import Union
+from typing import Union, Tuple
 
 import numpy as np
 from torch.utils import data
@@ -12,7 +12,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']