From c13f6e166d737a263fe096e209fe6d820ad5b37a Mon Sep 17 00:00:00 2001
From: Alex Waygood <Alex.Waygood@Gmail.com>
Date: Tue, 23 Apr 2024 12:26:02 +0100
Subject: [PATCH] Fix stubtest errors in tensorflow with `keras>=3.3.2`
 (#11817)

---
 stubs/tensorflow/METADATA.toml                        | 1 +
 stubs/tensorflow/tensorflow/keras/layers/__init__.pyi | 1 +
 2 files changed, 2 insertions(+)

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,