From 09a9dcf3f857e912c099806bece452bcf96722cb Mon Sep 17 00:00:00 2001
From: andreas128 <Andreas>
Date: Wed, 27 Sep 2017 17:45:12 +0200
Subject: Cleanup

---
 dpd/src/Model_Poly.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'dpd/src/Model_Poly.py')

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)
-- 
cgit v1.2.3