Add suffix to wrapper generated

plasma-umass · Jan 29, 2024 · e1c2e3a · e1c2e3a
1 parent aebbe49
commit e1c2e3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cwhy/__main__.py b/src/cwhy/__main__.py
@@ -165,7 +165,7 @@ def main():
     args = parser.parse_args()
 
     if args.wrapper:
-        with tempfile.NamedTemporaryFile(mode="w", delete=False) as f:
+        with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".py") as f:
             f.write(wrapper(args))
         # NamedTemporaryFiles are not executable by default.
         # Set its mode to 755 here with an octal literal.