From edaf4d220364854db1024483604ac6b83cf8f1ab Mon Sep 17 00:00:00 2001
From: su530201 <olivier.kaufmann@umons.ac.be>
Date: Wed, 18 Oct 2023 22:27:39 +0200
Subject: [PATCH] Fixes bug in _compute_tx_volt

---
 ohmpi/hardware_system.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ohmpi/hardware_system.py b/ohmpi/hardware_system.py
index 1819e7d3..fa53c1b6 100644
--- a/ohmpi/hardware_system.py
+++ b/ohmpi/hardware_system.py
@@ -478,9 +478,11 @@ class OhmPiHardware:
             switch_pwr_off = True
         k = 0
         diff_vab = np.inf
+
         while (k < n_steps) and (diff_vab > diff_vab_lim):
             vabs = []
             for pol in polarities:
+                vab = np.zeros(n_steps + 1) * np.nan
                 # self.tx.polarity = pol
                 # set gains automatically
                 injection = Thread(target=self._inject, kwargs={'injection_duration': 0.2, 'polarity': pol})
@@ -492,6 +494,7 @@ class OhmPiHardware:
                 injection.join()
                 v = np.where((self.readings[:, 0] > delay) & (self.readings[:, 2] != 0))[0] # NOTE : discard data aquired in the first x ms
                 iab = self.readings[v, 3]
+
                 vmn = self.readings[v, 4] * self.readings[v, 2]
                 iab_mean = np.mean(iab)
                 iab_std = np.std(iab)
-- 
GitLab