chore: hax

kszucs · Aug 6, 2024 · a9886a5 · a9886a5
1 parent 9f4cfc5
commit a9886a5
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/koerce/patterns.py b/koerce/patterns.py
@@ -1222,19 +1222,16 @@ class Capture(Pattern):
 
     """
 
-    key: str
+    key: str | Deferred | Builder
     what: Pattern
 
-    def __init__(self, k: str | Deferred | Builder, what=_any):
-        key: str
-        if isinstance(k, (Deferred, Builder)):
-            k = builder(k)
-            if isinstance(k, Variable):
-                key = k.name
+    def __init__(self, key: str | Deferred | Builder, what=_any):
+        if isinstance(key, (Deferred, Builder)):
+            key = builder(key)
+            if isinstance(key, Variable):
+                key = key.name
             else:
                 raise TypeError("Only variables can be used as capture keys")
-        else:
-            key = k
         self.key = key
         self.what = pattern(what)