fix context manager

tusharsadhwani · May 15, 2024 · 36d7208 · 36d7208
1 parent d3111d0
commit 36d7208
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/end_to_end/packaged_test.py b/tests/end_to_end/packaged_test.py
@@ -18,11 +18,16 @@ def build_package(
     output_path: str,
     build_command: str,
     startup_command: str,
+    python_version: str,
 ) -> Iterator[None]:
     """Builds the package, but also delete it afterwards."""
     try:
         packaged.create_package(
-            source_directory, output_path, build_command, startup_command
+            source_directory,
+            output_path,
+            build_command,
+            startup_command,
+            python_version,
         )
         yield
     finally: