Write exports to subdir in tests.

Listing the top level dir is not stable across OSes. PiperOrigin-RevId: 268556626
tensorflow · Sep 12, 2019 · 77199f9 · 77199f9
1 parent e081336
commit 77199f9
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/adanet/core/estimator_test.py b/adanet/core/estimator_test.py
@@ -1071,13 +1071,14 @@ def serving_input_fn():
     export_saved_model_fn = getattr(estimator, "export_saved_model", None)
     if not callable(export_saved_model_fn):
       export_saved_model_fn = estimator.export_savedmodel
+    export_dir_base = os.path.join(self.test_subdirectory, "export")
     export_saved_model_fn(
-        export_dir_base=self.test_subdirectory,
+        export_dir_base=export_dir_base,
         serving_input_receiver_fn=serving_input_fn)
     if export_subnetworks:
       saved_model = saved_model_utils.read_saved_model(
-          os.path.join(self.test_subdirectory,
-                       tf.io.gfile.listdir(self.test_subdirectory)[0]))
+          os.path.join(export_dir_base,
+                       tf.io.gfile.listdir(export_dir_base)[0]))
       export_signature_def = saved_model.meta_graphs[0].signature_def
       self.assertIn("subnetwork_logits", export_signature_def.keys())
       self.assertIn("subnetwork_last_layer", export_signature_def.keys())
@@ -1534,13 +1535,14 @@ def serving_input_fn():
     export_saved_model_fn = getattr(estimator, "export_saved_model", None)
     if not callable(export_saved_model_fn):
       export_saved_model_fn = estimator.export_savedmodel
+    export_dir_base = os.path.join(self.test_subdirectory, "export")
     export_saved_model_fn(
-        export_dir_base=self.test_subdirectory,
+        export_dir_base=export_dir_base,
         serving_input_receiver_fn=serving_input_fn)
     if export_subnetworks:
       saved_model = saved_model_utils.read_saved_model(
-          os.path.join(self.test_subdirectory,
-                       tf.io.gfile.listdir(self.test_subdirectory)[0]))
+          os.path.join(export_dir_base,
+                       tf.io.gfile.listdir(export_dir_base)[0]))
       export_signature_def = saved_model.meta_graphs[0].signature_def
       self.assertIn("subnetwork_logits_head1", export_signature_def.keys())
       self.assertIn("subnetwork_logits_head2", export_signature_def.keys())