Rename dummy calss not to use test prefix (quic#3717)

* Rename dummy class for testing without test prefix

---------

Signed-off-by: Hyeonwoo Koo <[email protected]>

TrainingExtensions/tensorflow/test/python/test_model_preparer.py

@@ -233,9 +233,9 @@ def call(self, inputs, **kwargs):
 
 
 # Layer with multiple inputs
-class TestMultiInput(tf.keras.layers.Layer):
+class MultiInputLayer(tf.keras.layers.Layer):
     def __init__(self, **kwargs) -> None:
-        super().__init__(name="test_multi_input")
+        super().__init__(name="multi_input_layer")
 
     def call(self, inputs, **kwargs):
         input1 = inputs[0]
@@ -244,9 +244,9 @@ def call(self, inputs, **kwargs):
 
 
 # Layer with multiple outputs
-class TestMultiOut(tf.keras.layers.Layer):
+class MultiOutLayer(tf.keras.layers.Layer):
     def __init__(self, **kwargs) -> None:
-        super().__init__(name="test_multi_out")
+        super().__init__(name="multi_out_layer")
 
     def call(self, inputs, **kwargs):
         out1 = inputs * 2.0
@@ -335,7 +335,7 @@ def get_model_with_multiple_inputs():
         input2 = tf.keras.Input(shape=(28, 28, 1), name="input_2")
         x = tf.keras.layers.Conv2D(16, 3, activation="relu")(input1)
         y = tf.keras.layers.Conv2D(16, 3, activation="relu")(input2)
-        outputs = TestMultiInput()([x, y])
+        outputs = MultiInputLayer()([x, y])
 
         return tf.keras.Model(inputs=[input1, input2], outputs=outputs, name="multi_input")
 
@@ -348,7 +348,7 @@ def get_model_with_multiple_outputs(use_lambdas):
             x1a = x1 * 2.0
             outputs = tf.concat([x0, x1a], 3)
         else:
-            outputs = TestMultiOut()(x)
+            outputs = MultiOutLayer()(x)
 
         return tf.keras.Model(inputs=inputs, outputs=outputs, name=f"multi_output_with_lambda_{use_lambdas}")