From 084c915c2ddb7552810a2222e64be49ab887ff5c Mon Sep 17 00:00:00 2001 From: "Yuan (Terry) Tang" Date: Fri, 1 Jul 2016 10:37:05 -0500 Subject: [PATCH] Fix arithmetic transform test (#3134) --- .../learn/tests/dataframe/arithmetic_transform_test.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/tensorflow/contrib/learn/python/learn/tests/dataframe/arithmetic_transform_test.py b/tensorflow/contrib/learn/python/learn/tests/dataframe/arithmetic_transform_test.py index be8305e3da3..2266caeb2f0 100644 --- a/tensorflow/contrib/learn/python/learn/tests/dataframe/arithmetic_transform_test.py +++ b/tensorflow/contrib/learn/python/learn/tests/dataframe/arithmetic_transform_test.py @@ -20,16 +20,24 @@ from __future__ import division from __future__ import print_function import numpy as np -import pandas as pd import tensorflow as tf from tensorflow.contrib.learn.python.learn.dataframe import tensorflow_dataframe as df +# pylint: disable=g-import-not-at-top +try: + import pandas as pd + HAS_PANDAS = True +except ImportError: + HAS_PANDAS = False + class SumTestCase(tf.test.TestCase): """Test class for `Sum` transform.""" def testSum(self): + if not HAS_PANDAS: + return num_rows = 100 pandas_df = pd.DataFrame({"a": np.arange(num_rows),