Fix condition for checking min_size in ResizeEdge class

datasets/augmentations.py

@@ -194,7 +194,7 @@ def __init__(
         self.min_size = min_size
         self.max_size = max_size
         self.edge_length = edge_length
-        if len(min_size) == 2 and min_size[0] == min_size[1]:
+        if set(min_size) == 1:
             self.edge_length = min_size[0]
 
     @override