summaryrefslogtreecommitdiffstats
path: root/dpd/src/Model_PM.py
diff options
context:
space:
mode:
Diffstat (limited to 'dpd/src/Model_PM.py')
-rw-r--r--dpd/src/Model_PM.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/dpd/src/Model_PM.py b/dpd/src/Model_PM.py
index 8d35051..eab5ec5 100644
--- a/dpd/src/Model_PM.py
+++ b/dpd/src/Model_PM.py
@@ -77,8 +77,10 @@ class Model_PM:
plt.close(fig)
def _discard_small_values(self, tx_dpd, phase_diff):
- mask = tx_dpd > self.c.MPM_tx_min
- return tx_dpd[mask], phase_diff[mask]
+ """ Assumes that the phase for small tx amplitudes is zero"""
+ mask = tx_dpd < self.c.MPM_tx_min
+ phase_diff[mask] = 0
+ return tx_dpd, phase_diff
def poly(self, sig):
return np.array([sig ** i for i in range(0, 5)]).T