From cd89309f6cfe1d1a5e5b6267d86be99cfc7995e3 Mon Sep 17 00:00:00 2001
From: Aditya Garg <85610623+AdityaGarg8@users.noreply.github.com>
Date: Mon, 2 Oct 2023 13:53:53 +0530
Subject: [PATCH] Firmware: Fix wifi script

---
 docs/tools/firmware.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/docs/tools/firmware.sh b/docs/tools/firmware.sh
index a250790d..1c534074 100644
--- a/docs/tools/firmware.sh
+++ b/docs/tools/firmware.sh
@@ -290,7 +290,9 @@ class WiFiFWCollection(object):
         self.prune()
 
     def load(self, source_path):
+    	included_folders = ["C-4355__s-C1", "C-4364__s-B2", "C-4364__s-B3", "C-4377__s-B3"]
         for dirpath, dirnames, filenames in os.walk(source_path):
+            dirnames[:] = [d for d in dirnames if d in included_folders]
             if "perf" in dirnames:
                 dirnames.remove("perf")
             if "assert" in dirnames: