diff options
Diffstat (limited to 'dpd/src/Model_Poly.py')
| -rw-r--r-- | dpd/src/Model_Poly.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/dpd/src/Model_Poly.py b/dpd/src/Model_Poly.py index 6a74bea..78bab91 100644 --- a/dpd/src/Model_Poly.py +++ b/dpd/src/Model_Poly.py @@ -62,7 +62,11 @@ class Poly: return self.coefs_am, self.coefs_pm def train(self, tx_abs, rx_abs, phase_diff): - # type: (np.ndarray, np.ndarray, np.ndarray) -> (str, np.ndarray, np.ndarray) + """ + :type tx_abs: np.ndarray + :type rx_abs: np.ndarray + :type phase_diff: np.ndarray + """ _check_input_get_next_coefs(tx_abs, rx_abs, phase_diff) coefs_am_new = self.model_am.get_next_coefs(tx_abs, rx_abs, self.coefs_am) |