Minor fixes in KMT initialization (#26402)

DataDog · Jul 19, 2024 · 0dfae31 · 0dfae31
1 parent 6f47983
commit 0dfae31
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tasks/kernel_matrix_testing/init_kmt.py b/tasks/kernel_matrix_testing/init_kmt.py
@@ -98,7 +98,7 @@ def init_kernel_matrix_testing_system(ctx: Context, lite: bool, images):
     # download dependencies
     if not lite:
         info("[+] Downloading VM images")
-        download_rootfs(ctx, kmt_os.rootfs_dir, "system-probe", images)
+        download_rootfs(ctx, kmt_os.rootfs_dir, "system-probe", arch=None, images=images)
 
     # Copy the SSH key we use to connect
     gen_ssh_key(ctx, kmt_os.kmt_dir)

diff --git a/tasks/kmt.py b/tasks/kmt.py
@@ -262,7 +262,7 @@ def init(ctx: Context, lite=False, images: str | None = None):
         init_kernel_matrix_testing_system(ctx, lite, images)
     except Exception as e:
         error(f"[-] Error initializing kernel matrix testing system: {e}")
-        raise Exit("[-] Initialization failed")
+        raise e
 
     info("[+] Kernel matrix testing system initialized successfully")
     config_ssh_key(ctx)