add python3 back

Author: yebrahim

.travis.yml

@@ -3,11 +3,11 @@ python:
   - "2.7"
 
 before_install:
-  - sudo apt-get install -y python-setuptools
+  - if [[ $(python -V 2>&1) == *"2.7"* ]]; then sudo apt-get install -y python-setuptools; fi
   - npm install -g typescript
   - tsc --module amd --noImplicitAny --outdir datalab/notebook/static datalab/notebook/static/*.ts
   - pip install -U pip
-  - pip install -U setuptools
+  - if [[ $(python -V 2>&1) == *"2.7"* ]]; then pip install -U setuptools; fi
   - pip install .
 
 script: python ./tests/main.py

setup.py

@@ -13,9 +13,12 @@
 # To publish to PyPi use: python setup.py bdist_wheel upload -r pypi
 
 import datetime
-import platform
+import sys
 from setuptools import setup
-import pip
+
+if sys.version_info[0] == 2:
+  import platform
+  import pip
 
 minor = datetime.datetime.now().strftime("%y%m%d%H%M")
 version = '0.1.' + minor
@@ -107,21 +110,23 @@
   }
 )
 
-tensorflow_path = None
-if platform.system() == 'Darwin':
-  tensorflow_path = 'https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-0.11.0-py2-none-any.whl'
-elif platform.system() == 'Linux':
-  if platform.linux_distribution()[0] == 'Ubuntu':
-    tensorflow_path = 'https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl'
-  elif platform.linux_distribution()[0] == 'Debian':
-    tensorflow_path = 'https://storage.googleapis.com/tensorflow/linux/cpu/debian/jessie/tensorflow-0.11.0-cp27-none-linux_x86_64.whl'
+# for python2 only, install tensorflow and cloudml
+if sys.version_info[0] == 2:
+  tensorflow_path = None
+  if platform.system() == 'Darwin':
+    tensorflow_path = 'https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-0.11.0-py2-none-any.whl'
+  elif platform.system() == 'Linux':
+    if platform.linux_distribution()[0] == 'Ubuntu':
+      tensorflow_path = 'https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl'
+    elif platform.linux_distribution()[0] == 'Debian':
+      tensorflow_path = 'https://storage.googleapis.com/tensorflow/linux/cpu/debian/jessie/tensorflow-0.11.0-cp27-none-linux_x86_64.whl'
 
-# install tensorflow
-if not tensorflow_path:
-  print("""Warning: could not find tensorflow build for your OS.
-  Please go to https://www.tensorflow.org/get_started/os_setup to see install options""")
-else:
-  pip.main(['install', tensorflow_path])
+  # install tensorflow
+  if not tensorflow_path:
+    print("""Warning: could not find tensorflow build for your OS.
+    Please go to https://www.tensorflow.org/get_started/os_setup to see install options""")
+  else:
+    pip.main(['install', tensorflow_path])
 
-# install cloud ml sdk
-pip.main(['install', 'https://storage.googleapis.com/cloud-ml/sdk/cloudml.latest.tar.gz'])
+  # install cloud ml sdk
+  pip.main(['install', 'https://storage.googleapis.com/cloud-ml/sdk/cloudml.latest.tar.gz'])