[set_linter] allow x in {...} (#148422)

jansel · pytorchmergebot · commit 85fe576ee3ec · 2025-03-09T06:43:11.000Z
Pull Request resolved: #148422 Approved by: https://github.com/Skylion007
diff --git a/tools/linter/adapters/set_linter.py b/tools/linter/adapters/set_linter.py
@@ -134,6 +134,8 @@ def is_set(self, i: int) -> bool:
     def is_braced_set(self, begin: int, end: int) -> bool:
         if begin + 1 == end or self.tokens[begin].string != "{":
             return False
+        if begin and self.tokens[begin - 1].string == "in":
+            return False  # skip `x in {1, 2, 3}`
         i = begin + 1
         empty = True
         while i < end:
