Fix stubtest errors in tensorflow with keras>=3.3.2 (#11817)

This commit is contained in:
Alex Waygood
2024-04-23 12:26:02 +01:00
committed by GitHub
parent af88af1fa6
commit c13f6e166d
2 changed files with 2 additions and 0 deletions

View File

@@ -9,3 +9,4 @@ partial_stub = true
[tool.stubtest]
ignore_missing_stub = true
stubtest_requirements = ["keras>=3.3.2"]

View File

@@ -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,