fix Effects.__iter__() for python3.10

vyperlang · Oct 16, 2024 · 1c98c23 · 1c98c23
1 parent ef84d70
commit 1c98c23
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/vyper/venom/effects.py b/vyper/venom/effects.py
@@ -12,6 +12,11 @@ class Effects(Flag):
     BALANCE = auto()
     EXTCODE = auto()
 
+    def __iter__(self):
+        # python3.10 doesn't have an iter implementation. we can
+        # remove this once we drop python3.10 support.
+        return (m for m in self.__class__.__members__.values() if m in self)
+
 
 EMPTY = Effects(0)
 ALL = ~EMPTY