From f4ce7e317a23c863751d3d8aeb2b94ff8edfd282 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Tue, 19 Nov 2019 16:24:24 +0800
Subject: [PATCH] Fix strip of socket message.

---
 paddleslim/common/controller_client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddleslim/common/controller_client.py b/paddleslim/common/controller_client.py
index fd6575e0..ad989dd1 100644
--- a/paddleslim/common/controller_client.py
+++ b/paddleslim/common/controller_client.py
@@ -51,7 +51,7 @@ class ControllerClient(object):
         socket_client.send("{}\t{}\t{}".format(self._key, tokens, reward)
                            .encode())
         response = socket_client.recv(1024).decode()
-        if response.trip('\n').split("\t") == "ok":
+        if response.strip('\n').split("\t") == "ok":
             return True
         else:
             return False
-- 
GitLab