From cbf4ad824417bedbb8caa41fc502192bffe5d9d5 Mon Sep 17 00:00:00 2001
From: Peter Izsak <peteriz@users.noreply.github.com>
Date: Wed, 28 Nov 2018 17:45:23 +0200
Subject: [PATCH] Peter/mkl install fix (#341)

* fix tf mkl install
---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 43510793..fd53ab6d 100755
--- a/setup.py
+++ b/setup.py
@@ -96,7 +96,8 @@
     tf_be = os.getenv('NLP_ARCHITECT_BE', False)
     if tf_be and 'mkl' == tf_be.lower():
         if py3_ver == 5 or py3_ver == 6:
-            chosen_tf = tf_mkl_url.format(tf_version, py3_ver, py3_ver)
+            tf_mkl_url_real = tf_mkl_url.format(tf_version, py3_ver, py3_ver)
+            subprocess.run('pip3 install -U {}'.format(tf_mkl_url_real), shell=True)
     elif tf_be and 'gpu' == tf_be.lower() and gpu_available:
         chosen_tf = 'tensorflow-gpu=={}'.format(tf_version)
 requirements.append(chosen_tf)