From 8b73c0f15961709c3e15ade819518fe9c6db11b2 Mon Sep 17 00:00:00 2001 From: Hyeonwoo Koo Date: Thu, 9 Jan 2025 09:28:37 +0900 Subject: [PATCH] Rename dummy calss not to use test prefix (#3717) * Rename dummy class for testing without test prefix --------- Signed-off-by: Hyeonwoo Koo --- .../tensorflow/test/python/test_model_preparer.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/TrainingExtensions/tensorflow/test/python/test_model_preparer.py b/TrainingExtensions/tensorflow/test/python/test_model_preparer.py index 9158f0055aa..96f4d66af81 100644 --- a/TrainingExtensions/tensorflow/test/python/test_model_preparer.py +++ b/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}")