diff --git a/examples/src/main/python/ml/als_example.py b/examples/src/main/python/ml/als_example.py index 922173308c6aa45076c3a535fd891e4109869300..0c9ac583b2dab568e32f82cf230e442c2e536a32 100644 --- a/examples/src/main/python/ml/als_example.py +++ b/examples/src/main/python/ml/als_example.py @@ -17,6 +17,10 @@ from __future__ import print_function +import sys +if sys.version >= '3': + long = int + from pyspark import SparkContext from pyspark.sql import SQLContext diff --git a/examples/src/main/python/mllib/gaussian_mixture_model.py b/examples/src/main/python/mllib/gaussian_mixture_model.py index 69e836fc1d06ae74eb33b4c82d98f80b8c39803f..6b46e27ddaaa8cfaf3d421e888f0b052de5f135f 100644 --- a/examples/src/main/python/mllib/gaussian_mixture_model.py +++ b/examples/src/main/python/mllib/gaussian_mixture_model.py @@ -20,6 +20,10 @@ A Gaussian Mixture Model clustering program using MLlib. """ from __future__ import print_function +import sys +if sys.version >= '3': + long = int + import random import argparse import numpy as np