fix versions parsing RegExp

eth-brownie · Feb 19, 2024 · dbd9658 · dbd9658
1 parent 405a3e4
commit dbd9658
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/brownie/project/sources.py b/brownie/project/sources.py
@@ -231,7 +231,9 @@ def get_vyper_pragma_spec(source: str, path: Optional[str] = None) -> NpmSpec:
 
     Returns: NpmSpec object
     """
-    pragma_match = next(re.finditer(r"(?:\n|^)\s*#\s*@version\s*([^\n]*)", source), None)
+    pragma_match = next(
+        re.finditer(r"(?:\n|^)\s*#\s*(?:pragma version|@version)\s*([^\n]*)", source), None
+    )
     if pragma_match is None:
         if path:
             raise PragmaError(f"No version pragma in '{path}'")