diff --git a/stubs/tensorflow/METADATA.toml b/stubs/tensorflow/METADATA.toml
index 8d9abbc7fea5..ba3efad4f3a2 100644
--- a/stubs/tensorflow/METADATA.toml
+++ b/stubs/tensorflow/METADATA.toml
@@ -9,3 +9,4 @@ partial_stub = true
 
 [tool.stubtest]
 ignore_missing_stub = true
+stubtest_requirements = ["keras>=3.3.2"]
diff --git a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
index c1dd1956baa3..6331017265b1 100644
--- a/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
+++ b/stubs/tensorflow/tensorflow/keras/layers/__init__.pyi
@@ -291,6 +291,7 @@ class Embedding(Layer[tf.Tensor, tf.Tensor]):
         embeddings_regularizer: _Regularizer = None,
         embeddings_constraint: _Constraint = None,
         mask_zero: bool = False,
+        weights=None,
         lora_rank: int | None = None,
         *,
         input_length: int | None = None,