experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #24635 from yongtang:12312018-test_session

Browse Source 
PiperOrigin-RevId: 230810202
This commit is contained in:
TensorFlower Gardener 2019-01-24 16:32:31 -08:00
commit df88a8eb41
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/kernel_tests/boosted_trees/quantile_ops_test.py
View File

@ -145,7 +145,7 @@ class QuantileOpsTest(test_util.TensorFlowTestCase):
save_dir = os.path.join(self.get_temp_dir(), "save_restore") save_dir = os.path.join(self.get_temp_dir(), "save_restore")
save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash") save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
with self.test_session() as sess: with self.cached_session() as sess:
accumulator = boosted_trees_ops.QuantileAccumulator( accumulator = boosted_trees_ops.QuantileAccumulator(
num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0") num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0")
@ -164,7 +164,7 @@ class QuantileOpsTest(test_util.TensorFlowTestCase):
self.assertAllClose(self._feature_1_boundaries, buckets[1].eval()) self.assertAllClose(self._feature_1_boundaries, buckets[1].eval())
save.save(sess, save_path) save.save(sess, save_path)
with self.test_session(graph=ops.Graph()) as sess: with self.session(graph=ops.Graph()) as sess:
accumulator = boosted_trees_ops.QuantileAccumulator( accumulator = boosted_trees_ops.QuantileAccumulator(
num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0") num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0")
save = saver.Saver() save = saver.Saver()
@ -177,7 +177,7 @@ class QuantileOpsTest(test_util.TensorFlowTestCase):
save_dir = os.path.join(self.get_temp_dir(), "save_restore") save_dir = os.path.join(self.get_temp_dir(), "save_restore")
save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash") save_path = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
with self.test_session() as sess: with self.cached_session() as sess:
accumulator = boosted_trees_ops.QuantileAccumulator( accumulator = boosted_trees_ops.QuantileAccumulator(
num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0") num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0")
@ -195,7 +195,7 @@ class QuantileOpsTest(test_util.TensorFlowTestCase):
self.assertAllClose(self._feature_0_boundaries, buckets[0].eval()) self.assertAllClose(self._feature_0_boundaries, buckets[0].eval())
self.assertAllClose(self._feature_1_boundaries, buckets[1].eval()) self.assertAllClose(self._feature_1_boundaries, buckets[1].eval())
with self.test_session(graph=ops.Graph()) as sess: with self.session(graph=ops.Graph()) as sess:
accumulator = boosted_trees_ops.QuantileAccumulator( accumulator = boosted_trees_ops.QuantileAccumulator(
num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0") num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0")
save = saver.Saver() save = saver.Saver()