From 9b065aad96d3df7e5791e7dcc81c9bab67eddc56 Mon Sep 17 00:00:00 2001
From: Stefan Klut <stefanklut12@gmail.com>
Date: Fri, 12 Apr 2024 12:19:35 +0200
Subject: [PATCH] Fix condition for checking min_size in ResizeEdge class

---
 datasets/augmentations.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/datasets/augmentations.py b/datasets/augmentations.py
index 7387655..0f80dcd 100644
--- a/datasets/augmentations.py
+++ b/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