From fd2eb432697f3cbb1aed941ca821753a1778584b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kier=C3=A1n=20Meinhardt?=
 <kieran.meinhardt@applicative.systems>
Date: Tue, 24 Mar 2026 15:11:23 +0100
Subject: [PATCH] nixos/test-driver: provide QemuMachine, NspawnMachine types
 to testScript

---
 nixos/lib/test-driver/src/test_driver/driver.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nixos/lib/test-driver/src/test_driver/driver.py b/nixos/lib/test-driver/src/test_driver/driver.py
index a457f7ed1be3..67c0172c3f5d 100644
--- a/nixos/lib/test-driver/src/test_driver/driver.py
+++ b/nixos/lib/test-driver/src/test_driver/driver.py
@@ -244,6 +244,8 @@ class Driver:
             serial_stdout_on=self.serial_stdout_on,
             polling_condition=self.polling_condition,
             BaseMachine=BaseMachine,  # for typing
+            QemuMachine=QemuMachine,  # for typing
+            NspawnMachine=NspawnMachine,  # for typing
             t=AssertionTester(),
             debug=self.debug,
         )