Revert "qemu: Remove unused argument from finalize_drive()"

This reverts commit 3e850db2bb5b4ad93742b844bb3f74f2c61f8b62.

diff --git a/mkosi/qemu.py b/mkosi/qemu.py
index 329133d62fa05883892bc8706d1225501b865e2c..dac6a9e80b989537b8784520d1d6e45322a284bf 100644 (file)
--- a/mkosi/qemu.py
+++ b/mkosi/qemu.py
@@ -795,7 +795,7 @@ def apply_runtime_size(config: Config, image: Path) -> None:
 
 
 @contextlib.contextmanager
-def finalize_drive(drive: Drive) -> Iterator[Path]:
+def finalize_drive(config: Config, drive: Drive) -> Iterator[Path]:
     with contextlib.ExitStack() as stack:
         file: IO[bytes]
         if DriveFlag.persist in drive.flags:
@@ -1533,7 +1533,7 @@ def run_qemu(args: Args, config: Config) -> None:
             ]
 
         for _, drives in groupby(config.drives, key=lambda d: d.file_id):
-            file = stack.enter_context(finalize_drive(drives[0]))
+            file = stack.enter_context(finalize_drive(config, drives[0]))
 
             for drive in drives:
                 arg = [