Move one time use setup function

googleapis · Jul 8, 2021 · 7ceb25d · 7ceb25d
1 parent aa37d3b
commit 7ceb25d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_test.py b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_test.py
@@ -23,16 +23,15 @@
 
 
 class SavedModelMetadataBuilderTest(tf.test.TestCase):
-    def _set_up(self):
+    def test_get_metadata_sequential(self):
+        # Set up for the sequential.
         self.seq_model = tf.keras.models.Sequential()
         self.seq_model.add(tf.keras.layers.Dense(32, activation="relu", input_dim=10))
         self.seq_model.add(tf.keras.layers.Dense(32, activation="relu"))
         self.seq_model.add(tf.keras.layers.Dense(1, activation="sigmoid"))
         self.saved_model_path = self.get_temp_dir()
         tf.saved_model.save(self.seq_model, self.saved_model_path)
 
-    def test_get_metadata_sequential(self):
-        self._set_up()
         builder = saved_model_metadata_builder.SavedModelMetadataBuilder(
             self.saved_model_path
         )