From 5106276e7c5757b98f4045f573bcddc949cbdc9a Mon Sep 17 00:00:00 2001
From: su530201 <olivier.kaufmann@umons.ac.be>
Date: Tue, 2 May 2023 18:02:17 +0200
Subject: [PATCH] Refactors mux barrier to only activate muxes involved

---
 hardware_system.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/hardware_system.py b/hardware_system.py
index fe1e0170..0c60657a 100644
--- a/hardware_system.py
+++ b/hardware_system.py
@@ -284,9 +284,12 @@ class OhmPiHardware:
             mux_workers = []
             for idx, elec in enumerate(electrodes):
                 elec_dict[roles[idx]].append(elec)
-                mux = self._cabling[(elec, roles[idx])][0]
-                if mux not in mux_workers:
-                    mux_workers.append(mux)
+                try:
+                    mux = self._cabling[(elec, roles[idx])][0]
+                    if mux not in mux_workers:
+                        mux_workers.append(mux)
+                except KeyError:
+                    self.exec_logger.warning(f'({elec}, {roles[idx]} is not in cabling. It will be ignored...')
             mux_workers = list(set(mux_workers))
             b = Barrier(len(mux_workers)+1)
             self.mux_barrier = b
-- 
GitLab