diff options
| -rw-r--r-- | examples/src/main/python/ml/als_example.py | 4 | ||||
| -rw-r--r-- | examples/src/main/python/mllib/gaussian_mixture_model.py | 4 |
2 files changed, 8 insertions, 0 deletions
diff --git a/examples/src/main/python/ml/als_example.py b/examples/src/main/python/ml/als_example.py index 922173308c..0c9ac583b2 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 69e836fc1d..6b46e27dda 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 |