diff options
Diffstat (limited to 'utils/sampler.py')
-rw-r--r-- | utils/sampler.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/utils/sampler.py b/utils/sampler.py index e609e2d..b017d66 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 |