From 47a1a003339b8682765cef2211e1a953725a4fa0 Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Wed, 22 Apr 2020 17:03:55 +0800
Subject: [PATCH] update run version judge

---
 fleetrec/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fleetrec/run.py b/fleetrec/run.py
index c7bc096a..984e1e19 100644
--- a/fleetrec/run.py
+++ b/fleetrec/run.py
@@ -52,7 +52,7 @@ def set_runtime_envs(cluster_envs, engine_yaml):
 
 def get_engine(engine):
     engine = engine.upper()
-    if version.is_transpiler():
+    if is_transpiler():
         run_engine = engines["TRAINSPILER"].get(engine, None)
     else:
         run_engine = engines["PSLIB"].get(engine, None)
-- 
GitLab