diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/estimator_test.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/estimator_test.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py b/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py index 855c44d518..cfb56a9d8d 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py +++ b/tensorflow/contrib/learn/python/learn/estimators/estimator_test.py @@ -505,7 +505,7 @@ class EstimatorModelFnTest(test.TestCase): return input_fn_utils.InputFnOps( features, labels, {'examples': serialized_tf_example}) - est.export_savedmodel(est.model_dir + '/export', serving_input_fn) + est.export_savedmodel(os.path.join(est.model_dir, 'export'), serving_input_fn) self.assertTrue(self.mock_saver.restore.called) @@ -955,10 +955,11 @@ class EstimatorTest(test.TestCase): self.assertTrue('input_example_tensor' in graph_ops) self.assertTrue('ParseExample/ParseExample' in graph_ops) self.assertTrue('linear/linear/feature/matmul' in graph_ops) - self.assertSameElements( - ['bogus_lookup', 'feature'], - graph.get_collection( - constants.COLLECTION_DEF_KEY_FOR_INPUT_FEATURE_KEYS)) + self.assertItemsEqual( + ['bogus_lookup', 'feature'], + [compat.as_str_any(x) for x in graph.get_collection( + constants.COLLECTION_DEF_KEY_FOR_INPUT_FEATURE_KEYS)]) + # cleanup gfile.DeleteRecursively(tmpdir) |