Skip to content

Commit c13f6e1

Browse files
authored
Fix stubtest errors in tensorflow with keras>=3.3.2 (#11817)
1 parent af88af1 commit c13f6e1

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

stubs/tensorflow/METADATA.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,3 +9,4 @@ partial_stub = true
99

1010
[tool.stubtest]
1111
ignore_missing_stub = true
12+
stubtest_requirements = ["keras>=3.3.2"]

stubs/tensorflow/tensorflow/keras/layers/__init__.pyi

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -291,6 +291,7 @@ class Embedding(Layer[tf.Tensor, tf.Tensor]):
291291
embeddings_regularizer: _Regularizer = None,
292292
embeddings_constraint: _Constraint = None,
293293
mask_zero: bool = False,
294+
weights=None,
294295
lora_rank: int | None = None,
295296
*,
296297
input_length: int | None = None,

0 commit comments

Comments
 (0)