From d8e008cceafc5b4e124d93d6a5119ace00acce39 Mon Sep 17 00:00:00 2001 From: mohantym <86464649+mohantym@users.noreply.github.com> Date: Mon, 11 Apr 2022 09:16:11 +0530 Subject: [PATCH] Removed extra comma after x_train Proposing to remove extra comma after x_train at 153 and 191 to avoid converting the value to tuple. --- .../python/core/sparsity/keras/prune_integration_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/prune_integration_test.py b/tensorflow_model_optimization/python/core/sparsity/keras/prune_integration_test.py index c7b443a77..f62691503 100644 --- a/tensorflow_model_optimization/python/core/sparsity/keras/prune_integration_test.py +++ b/tensorflow_model_optimization/python/core/sparsity/keras/prune_integration_test.py @@ -150,7 +150,7 @@ def _train_model(self, y_train=None, callbacks=None): if x_train is None: - x_train = np.random.rand(20, 10), + x_train = np.random.rand(20, 10) if y_train is None: y_train = keras.utils.to_categorical( np.random.randint(5, size=(20, 1)), 5) @@ -188,7 +188,7 @@ def testPrunesZeroSparsity_IsNoOp(self): target_sparsity=0, begin_step=0, frequency=1) pruned_model = prune.prune_low_magnitude(model2, **params) - x_train = np.random.rand(20, 10), + x_train = np.random.rand(20, 10) y_train = keras.utils.to_categorical(np.random.randint(5, size=(20, 1)), 5) self._train_model(model, epochs=1, x_train=x_train, y_train=y_train)