From cfca2d6cb54c4cf67831d5cc791509bfbbbe5c02 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Wed, 31 Dec 2014 12:37:51 +0200
Subject: [PATCH] equalizer: remove unused variables

---
 amodem/equalizer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/amodem/equalizer.py b/amodem/equalizer.py
index da85920..89df50f 100644
--- a/amodem/equalizer.py
+++ b/amodem/equalizer.py
@@ -21,7 +21,7 @@ class Equalizer(object):
     def train_symbols(self, length, seed=0):
         r = random.Random(seed)
         choose = lambda: [r.choice(_constellation) for j in range(self.Nfreq)]
-        return np.array([choose() for i in range(length)])
+        return np.array([choose() for _ in range(length)])
 
     def modulator(self, symbols):
         gain = 1.0 / len(self.carriers)
@@ -59,7 +59,7 @@ class Equalizer(object):
 
         A = np.array(A)
         b = np.array(b)
-        h, residuals, rank, sv = lstsq(A, b)
+        h = lstsq(A, b)[0]
         h = h[::-1].real
 
         return h