walrus operator was swallowing stuff

ApeWorX · Oct 24, 2023 · 8df6803 · 8df6803
1 parent 64e29e9
commit 8df6803
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ape_vyper/compiler.py b/ape_vyper/compiler.py
@@ -402,10 +402,10 @@ def get_optimization_pragma_map(
         base_path = base_path or self.config_manager.contracts_folder
         optimization_pragma_map: Dict[Union[str, bool], Set[Path]] = {}
         for path in contract_filepaths:
-            if pragma := get_optimization_pragma(path):
-                if pragma not in optimization_pragma_map:
-                    optimization_pragma_map[pragma] = set()
-                optimization_pragma_map[pragma].add(path)
+            pragma = get_optimization_pragma(path) or True
+            if pragma not in optimization_pragma_map:
+                optimization_pragma_map[pragma] = set()
+            optimization_pragma_map[pragma].add(path)
 
         return optimization_pragma_map