From 58cbc2a49b7bfac47e4245118d3d4e719aa46db7 Mon Sep 17 00:00:00 2001
From: Florent 'Skia' Jacquet <florent.jacquet@canonical.com>
Date: Wed, 26 Feb 2025 09:52:57 +0100
Subject: [PATCH] policies: fix traceback when trying to run riscv64
 autopkgtests

---
 britney2/policies/autopkgtest.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/britney2/policies/autopkgtest.py b/britney2/policies/autopkgtest.py
index ffefc0d..5089ef3 100644
--- a/britney2/policies/autopkgtest.py
+++ b/britney2/policies/autopkgtest.py
@@ -842,8 +842,12 @@ class AutopkgtestPolicy(BasePolicy):
                             break
 
         # Filter tests to main packages on riscv64
-        if arch == "riscv64":
-            tests = [(src, version) for (src, version) in tests if sources_info[src].component == "main"]
+        try:
+            # Filter tests to main packages on riscv64
+            if arch == "riscv64":
+                tests = [(src, version) for (src, version) in tests if sources_info[src].component == "main"]
+        except KeyError:  # Sometimes™, sources_info[src] raises KeyError
+            pass
 
         tests.sort(key=lambda s_v: s_v[0])
         return tests