Merge pull request #464 from cliveseldon/python-type-updates

Remove tornando dependency from Python setup.py
SeldonIO · Mar 13, 2019 · fe570bf · fe570bf
2 parents d4992e5 + 3fad42a
commit fe570bf
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 9 deletions.
diff --git a/python/seldon_core/api_tester.py b/python/seldon_core/api_tester.py
@@ -61,9 +61,7 @@ def run_send_feedback(args):
                                         prediction_response=response_predict.response, reward=1.0,
                                         deployment_name=args.deployment, transport=transport)
         if args.prnt:
-            print("RECEIVED RESPONSE:")
-            print(response_feedback)
-            print()
+            print(f"RECEIVED RESPONSE:\n{response_feedback}\n")
 
 
 def run_predict(args):
@@ -87,11 +85,12 @@ def run_predict(args):
 
     for i in range(args.n_requests):
         batch = generate_batch(contract, args.batch_size, 'features')
+        if args.prnt:
+            print(f"{'-' * 40}\nSENDING NEW REQUEST:\n")
+            print(batch)
         response_predict = sc.predict(data=batch, deployment_name=args.deployment)
         if args.prnt:
-            print("RECEIVED RESPONSE:")
-            print(response_predict)
-            print()
+            print(f"RECEIVED RESPONSE:\n{response_predict.response}\n")
 
 
 def main():

diff --git a/python/seldon_core/microservice_tester.py b/python/seldon_core/microservice_tester.py
@@ -204,15 +204,16 @@ def run_predict(args):
     for i in range(args.n_requests):
         batch: ndarray = generate_batch(contract, args.batch_size, 'features')
         if args.prnt:
-            print(f"{'-' * 40}\nSENDING NEW REQUEST:")
+            print(f"{'-' * 40}\nSENDING NEW REQUEST:\n")
+            print(batch)
 
         transport = "grpc" if args.grpc else "rest"
         payload_type = "tensor" if args.tensor else "ndarray"
 
         response = sc.microservice(data=batch, transport=transport, method="predict", payload_type=payload_type)
 
         if args.prnt:
-            print(f"RECEIVED RESPONSE:\n{response}\n")
+            print(f"RECEIVED RESPONSE:\n{response.response}\n")
 
 
 def main():

diff --git a/python/setup.py b/python/setup.py
@@ -23,7 +23,6 @@
           'flask',
           'flask-cors',
           'redis',
-          'tornado>=4.3,<5',
           'requests',
           'numpy',
           'flatbuffers',