From 3f8d8f8c2f13abaf6d64da3616b8bc593bff505a Mon Sep 17 00:00:00 2001 From: Yang Wang Date: Mon, 24 Aug 2020 19:44:30 +0800 Subject: [PATCH] fix test style (#2770) --- .../orca/test/bigdl/orca/learn/ray/tf/test_tf_ray_estimator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/orca/test/bigdl/orca/learn/ray/tf/test_tf_ray_estimator.py b/python/orca/test/bigdl/orca/learn/ray/tf/test_tf_ray_estimator.py index 5127673dcb1..15ddd998adf 100644 --- a/python/orca/test/bigdl/orca/learn/ray/tf/test_tf_ray_estimator.py +++ b/python/orca/test/bigdl/orca/learn/ray/tf/test_tf_ray_estimator.py @@ -59,7 +59,8 @@ def create_test_dataset(config): def simple_model(config): import tensorflow as tf - model = tf.keras.models.Sequential([tf.keras.layers.Dense(10, input_shape=(1,)), tf.keras.layers.Dense(1)]) + model = tf.keras.models.Sequential([tf.keras.layers.Dense(10, input_shape=(1,)), + tf.keras.layers.Dense(1)]) return model