Fix Value error for Units of tf.keras.layers.SimpleRNN
@@ -121,7 +121,7 @@ class SimpleRNNCell(DropoutRNNCellMixin, base_layer.BaseRandomLayer):
recurrent_dropout=0.0,
**kwargs,
):
- if units < 0:
+ if units <= 0:
raise ValueError(
"Received an invalid value for argument `units`, "
f"expected a positive integer, got {units}."
