diff options
author | 2018-09-10 14:36:52 -0700 | |
---|---|---|
committer | 2018-09-10 15:01:47 -0700 | |
commit | acf0ee82092727afc2067316982407cf5e496f75 (patch) | |
tree | e5df1811ab47e259a1f30c46e22c251411ad326e /tensorflow/contrib/learn | |
parent | f1cc58bb4144de61a693076d8ff8a26b2644ebbb (diff) |
Move from deprecated self.test_session() to self.cached_session().
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 212336417
Diffstat (limited to 'tensorflow/contrib/learn')
3 files changed, 24 insertions, 24 deletions
diff --git a/tensorflow/contrib/learn/python/learn/learn_io/data_feeder_test.py b/tensorflow/contrib/learn/python/learn/learn_io/data_feeder_test.py index 5e07b9313f..284a4f45f6 100644 --- a/tensorflow/contrib/learn/python/learn/learn_io/data_feeder_test.py +++ b/tensorflow/contrib/learn/python/learn/learn_io/data_feeder_test.py @@ -147,7 +147,7 @@ class DataFeederTest(test.TestCase): def test_unsupervised(self): def func(feeder): - with self.test_session(): + with self.cached_session(): inp, _ = feeder.input_builder() feed_dict_fn = feeder.get_feed_dict_fn() feed_dict = feed_dict_fn() @@ -181,7 +181,7 @@ class DataFeederTest(test.TestCase): def test_epoch(self): def func(feeder): - with self.test_session(): + with self.cached_session(): feeder.input_builder() epoch = feeder.make_epoch_variable() feed_dict_fn = feeder.get_feed_dict_fn() diff --git a/tensorflow/contrib/learn/python/learn/learn_io/generator_io_test.py b/tensorflow/contrib/learn/python/learn/learn_io/generator_io_test.py index 7e81f2b7d9..5e90d1fa20 100644 --- a/tensorflow/contrib/learn/python/learn/learn_io/generator_io_test.py +++ b/tensorflow/contrib/learn/python/learn/learn_io/generator_io_test.py @@ -38,7 +38,7 @@ class GeneratorIoTest(test.TestCase): 'label': np.ones(1) * index - 32 } - with self.test_session() as session: + with self.cached_session() as session: input_fn = generator_io.generator_input_fn( generator, target_key='label', @@ -68,7 +68,7 @@ class GeneratorIoTest(test.TestCase): for index in range(2): yield {'a': np.ones(1) * index} - with self.test_session() as session: + with self.cached_session() as session: input_fn = generator_io.generator_input_fn( generator, target_key=None, batch_size=2, shuffle=False, num_epochs=1) features = input_fn() @@ -97,7 +97,7 @@ class GeneratorIoTest(test.TestCase): 'label2': np.ones(1) * index - 64, } - with self.test_session() as session: + with self.cached_session() as session: input_fn = generator_io.generator_input_fn( generator, target_key=['label', 'label2'], @@ -134,7 +134,7 @@ class GeneratorIoTest(test.TestCase): 'label': np.ones((3, 3)) * index - 32 } - with self.test_session() as session: + with self.cached_session() as session: input_fn = generator_io.generator_input_fn( generator, target_key='label', @@ -162,7 +162,7 @@ class GeneratorIoTest(test.TestCase): def testGeneratorInputFnWithXAsNonGeneratorFunction(self): x = np.arange(32, 36) - with self.test_session(): + with self.cached_session(): with self.assertRaisesRegexp(TypeError, 'x must be generator function'): failing_input_fn = generator_io.generator_input_fn( x, batch_size=2, shuffle=False, num_epochs=1) @@ -173,7 +173,7 @@ class GeneratorIoTest(test.TestCase): def generator(): return np.arange(32, 36) - with self.test_session(): + with self.cached_session(): with self.assertRaisesRegexp(TypeError, 'x\(\) must be generator'): failing_input_fn = generator_io.generator_input_fn( generator, batch_size=2, shuffle=False, num_epochs=1) @@ -184,7 +184,7 @@ class GeneratorIoTest(test.TestCase): def generator(): yield np.arange(32, 36) - with self.test_session(): + with self.cached_session(): with self.assertRaisesRegexp(TypeError, 'x\(\) must yield dict'): failing_input_fn = generator_io.generator_input_fn( generator, batch_size=2, shuffle=False, num_epochs=1) @@ -201,7 +201,7 @@ class GeneratorIoTest(test.TestCase): } y = np.arange(32, 36) - with self.test_session(): + with self.cached_session(): with self.assertRaisesRegexp(TypeError, 'target_key must be str or' ' Container of str'): failing_input_fn = generator_io.generator_input_fn( @@ -219,7 +219,7 @@ class GeneratorIoTest(test.TestCase): } y = ['label', np.arange(10)] - with self.test_session(): + with self.cached_session(): with self.assertRaisesRegexp(TypeError, 'target_key must be str or' ' Container of str'): failing_input_fn = generator_io.generator_input_fn( @@ -237,7 +237,7 @@ class GeneratorIoTest(test.TestCase): } y = ['label', 'target'] - with self.test_session(): + with self.cached_session(): with self.assertRaisesRegexp(KeyError, 'target_key not in yielded dict'): failing_input_fn = generator_io.generator_input_fn( generator, target_key=y, batch_size=2, shuffle=False, num_epochs=1) @@ -253,7 +253,7 @@ class GeneratorIoTest(test.TestCase): 'label': np.ones(1) * index - 32 } - with self.test_session() as session: + with self.cached_session() as session: input_fn = generator_io.generator_input_fn( generator, target_key=None, batch_size=2, shuffle=False, num_epochs=1) features = input_fn() @@ -283,7 +283,7 @@ class GeneratorIoTest(test.TestCase): 'label': np.ones(1) * index - 32 } - with self.test_session() as session: + with self.cached_session() as session: input_fn = generator_io.generator_input_fn( generator, target_key=None, batch_size=4, shuffle=False, num_epochs=1) features = input_fn() @@ -319,7 +319,7 @@ class GeneratorIoTest(test.TestCase): 'label': np.ones(1) * index - 32 } - with self.test_session() as session: + with self.cached_session() as session: input_fn = generator_io.generator_input_fn( generator, target_key=None, batch_size=2, shuffle=False, num_epochs=1) features = input_fn() diff --git a/tensorflow/contrib/learn/python/learn/learn_io/pandas_io_test.py b/tensorflow/contrib/learn/python/learn/learn_io/pandas_io_test.py index c738f0e8f3..396539a76a 100644 --- a/tensorflow/contrib/learn/python/learn/learn_io/pandas_io_test.py +++ b/tensorflow/contrib/learn/python/learn/learn_io/pandas_io_test.py @@ -65,7 +65,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_ProducesExpectedOutputs(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: x, y = self.makeTestDataFrame() input_fn = pandas_io.pandas_input_fn( x, y, batch_size=2, shuffle=False, num_epochs=1) @@ -79,7 +79,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_ProducesOutputsForLargeBatchAndMultipleEpochs(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: index = np.arange(100, 102) a = np.arange(2) b = np.arange(32, 34) @@ -107,7 +107,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_ProducesOutputsWhenDataSizeNotDividedByBatchSize(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: index = np.arange(100, 105) a = np.arange(5) b = np.arange(32, 37) @@ -146,7 +146,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_OnlyX(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: x, _ = self.makeTestDataFrame() input_fn = pandas_io.pandas_input_fn( x, y=None, batch_size=2, shuffle=False, num_epochs=1) @@ -159,7 +159,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_ExcludesIndex(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: x, y = self.makeTestDataFrame() input_fn = pandas_io.pandas_input_fn( x, y, batch_size=2, shuffle=False, num_epochs=1) @@ -182,7 +182,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_RespectsEpoch_NoShuffle(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: x, y = self.makeTestDataFrame() input_fn = pandas_io.pandas_input_fn( x, y, batch_size=4, shuffle=False, num_epochs=1) @@ -192,7 +192,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_RespectsEpoch_WithShuffle(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: x, y = self.makeTestDataFrame() input_fn = pandas_io.pandas_input_fn( x, y, batch_size=4, shuffle=True, num_epochs=1) @@ -202,7 +202,7 @@ class PandasIoTest(test.TestCase): def testPandasInputFn_RespectsEpoch_WithShuffleAutosize(self): if not HAS_PANDAS: return - with self.test_session() as session: + with self.cached_session() as session: x, y = self.makeTestDataFrame() input_fn = pandas_io.pandas_input_fn( x, y, batch_size=2, shuffle=True, queue_capacity=None, num_epochs=2) @@ -213,7 +213,7 @@ class PandasIoTest(test.TestCase): if not HAS_PANDAS: return x, y = self.makeTestDataFrame() - with self.test_session() as session: + with self.cached_session() as session: input_fn = pandas_io.pandas_input_fn( x, y, batch_size=3, shuffle=False, num_epochs=1) |