From 03f269c2d7ac1ceba9ea0663c2fdb75ced70865e Mon Sep 17 00:00:00 2001
From: Matthias Bisping <matthias.bisping@iqser.com>
Date: Fri, 25 Mar 2022 16:28:00 +0100
Subject: [PATCH] fixed incorrect pycharme-refactoring

---
 scripts/keras_MnWE.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/keras_MnWE.py b/scripts/keras_MnWE.py
index 0788e43..88b920a 100644
--- a/scripts/keras_MnWE.py
+++ b/scripts/keras_MnWE.py
@@ -26,7 +26,7 @@ def make_model():
     dense = layers.Dense(64, activation="relu")
     x = dense(inputs)
     outputs = layers.Dense(10)(x)
-    model = keras.ServiceEstimator(inputs=inputs, outputs=outputs, name="mnist_model")
+    model = keras.Model(inputs=inputs, outputs=outputs, name="mnist_model")
     model.compile(
         loss=keras.losses.SparseCategoricalCrossentropy(from_logits=True),
         optimizer=keras.optimizers.RMSprop(),