fix: typ hint

ApeWorX · Oct 24, 2023 · ae20d69 · ae20d69
1 parent 6cb728e
commit ae20d69
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/ape_vyper/compiler.py b/ape_vyper/compiler.py
@@ -100,7 +100,7 @@ def get_version_pragma_spec(source: Union[str, Path]) -> Optional[SpecifierSet]:
         return None
 
 
-def get_optimization_pragma(source: Union[str, Path]) -> Optional[str | bool]:
+def get_optimization_pragma(source: Union[str, Path]) -> str | bool | None:
     """
     Extracts optimization pragma information from Vyper source code.
 

diff --git a/tests/contracts/passing_contracts/optimize_codesize.vy b/tests/contracts/passing_contracts/optimize_codesize.vy
@@ -0,0 +1,8 @@
+#pragma version 0.3.10
+#pragma optimize codesize
+
+x: uint256
+
+@external
+def __init__():
+    self.x = 0