support install from venv and travis CI

microsoft · Sep 12, 2018 · 57cd372 · 57cd372
1 parent 6c1c3f3
commit 57cd372
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/Makefile b/Makefile
@@ -4,9 +4,6 @@ EXAMPLES_PATH ?= ${HOME}/nni/examples
 WHOAMI := $(shell whoami)
 YARN := $(INSTALL_PREFIX)/yarn/bin/yarn
 PIP_MODE ?= --user
-ifdef TRAVIS
-undefine PIP_MODE
-endif
 ifdef VIRTUAL_ENV
 undefine PIP_MODE
 endif

diff --git a/test/naive/run.py b/test/naive/run.py
@@ -76,10 +76,10 @@ def run():
         run()
         # TODO: check the output of rest server
         print(GREEN + 'PASS' + CLEAR)
-    except Exception as e:
+    except Exception as error:
         print(RED + 'FAIL' + CLEAR)
-        print('%r' % e)
+        print('%r' % error)
         traceback.print_exc()
-        raise e
+        raise error
 
     subprocess.run(['nnictl', 'stop'])