From f886f5377f6d942a092c8ce4a684dbd7ca2eddf5 Mon Sep 17 00:00:00 2001 From: nrstott Date: Mon, 2 Jul 2018 12:37:15 -0400 Subject: [PATCH] lint errors --- tensorflow/python/estimator/inputs/pandas_io.py | 2 +- .../python/estimator/inputs/pandas_io_test.py | 16 ++++++++-------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/tensorflow/python/estimator/inputs/pandas_io.py b/tensorflow/python/estimator/inputs/pandas_io.py index af8030701be..616bcb410f8 100644 --- a/tensorflow/python/estimator/inputs/pandas_io.py +++ b/tensorflow/python/estimator/inputs/pandas_io.py @@ -112,7 +112,7 @@ def pandas_input_fn(x, 'Index for y: %s\n' % (x.index, y.index)) if isinstance(y, pd.DataFrame): y_columns = [(column, _get_unique_target_key(x, column)) - for column in list(y)] + for column in list(y)] target_column = [v for _, v in y_columns] x[target_column] = y else: diff --git a/tensorflow/python/estimator/inputs/pandas_io_test.py b/tensorflow/python/estimator/inputs/pandas_io_test.py index 768b8c19fdc..6f13bc95d2d 100644 --- a/tensorflow/python/estimator/inputs/pandas_io_test.py +++ b/tensorflow/python/estimator/inputs/pandas_io_test.py @@ -84,9 +84,9 @@ class PandasIoTest(test.TestCase): with self.assertRaisesRegexp(TypeError, 'target_column must be a string type'): pandas_io.pandas_input_fn(x, y, batch_size=2, - shuffle=False, - num_epochs=1, - target_column=['one', 'two']) + shuffle=False, + num_epochs=1, + target_column=['one', 'two']) def testPandasInputFn_NonBoolShuffle(self): if not HAS_PANDAS: @@ -119,7 +119,7 @@ class PandasIoTest(test.TestCase): with self.test_session() as session: x, y = self.makeTestDataFrameWithYAsDataFrame() input_fn = pandas_io.pandas_input_fn( - x, y, batch_size=2, shuffle=False, num_epochs=1) + x, y, batch_size=2, shuffle=False, num_epochs=1) features, targets = self.callInputFnOnce(input_fn, session) @@ -128,14 +128,14 @@ class PandasIoTest(test.TestCase): self.assertAllEqual(targets['a_target'], [10, 11]) self.assertAllEqual(targets['b_target'], [50, 51]) - def testPandasInputFnWhenYIsDataFrame_HandlesOverlappingColumnNames(self): + def testPandasInputFnYIsDataFrame_HandlesOverlappingColumns(self): if not HAS_PANDAS: return with self.test_session() as session: x, y = self.makeTestDataFrameWithYAsDataFrame() y = y.rename(columns={'a_target': 'a', 'b_target': 'b'}) input_fn = pandas_io.pandas_input_fn( - x, y, batch_size=2, shuffle=False, num_epochs=1) + x, y, batch_size=2, shuffle=False, num_epochs=1) features, targets = self.callInputFnOnce(input_fn, session) @@ -144,14 +144,14 @@ class PandasIoTest(test.TestCase): self.assertAllEqual(targets['a'], [10, 11]) self.assertAllEqual(targets['b'], [50, 51]) - def testPandasInputFnWhenYIsDataFrame_HandlesOverlappingColumnNamesInTargets(self): + def testPandasInputFnYIsDataFrame_HandlesOverlappingColumnsInTargets(self): if not HAS_PANDAS: return with self.test_session() as session: x, y = self.makeTestDataFrameWithYAsDataFrame() y = y.rename(columns={'a_target': 'a', 'b_target': 'a_n'}) input_fn = pandas_io.pandas_input_fn( - x, y, batch_size=2, shuffle=False, num_epochs=1) + x, y, batch_size=2, shuffle=False, num_epochs=1) features, targets = self.callInputFnOnce(input_fn, session)