From 2db0b6e655ffab2fdb0365d8ec7a028a8095e56d Mon Sep 17 00:00:00 2001 From: Alex Waygood Date: Sat, 27 Apr 2024 09:42:34 +0100 Subject: [PATCH] Fix stubtest for TensorFlow with latest keras release (#11838) --- stubs/tensorflow/METADATA.toml | 2 +- stubs/tensorflow/tensorflow/keras/losses.pyi | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/stubs/tensorflow/METADATA.toml b/stubs/tensorflow/METADATA.toml index ba3efad4f..57c674a03 100644 --- a/stubs/tensorflow/METADATA.toml +++ b/stubs/tensorflow/METADATA.toml @@ -9,4 +9,4 @@ partial_stub = true [tool.stubtest] ignore_missing_stub = true -stubtest_requirements = ["keras>=3.3.2"] +stubtest_requirements = ["keras>=3.3.3"] diff --git a/stubs/tensorflow/tensorflow/keras/losses.pyi b/stubs/tensorflow/tensorflow/keras/losses.pyi index fc2881784..04ac8a527 100644 --- a/stubs/tensorflow/tensorflow/keras/losses.pyi +++ b/stubs/tensorflow/tensorflow/keras/losses.pyi @@ -142,7 +142,6 @@ _ReductionValues: TypeAlias = Literal["auto", "none", "sum", "sum_over_batch_siz def categorical_hinge(y_true: TensorCompatible, y_pred: TensorCompatible) -> Tensor: ... def huber(y_true: TensorCompatible, y_pred: TensorCompatible, delta: float = 1.0) -> Tensor: ... -def log_cosh(y_true: TensorCompatible, y_pred: TensorCompatible) -> Tensor: ... def deserialize(name: str | dict[str, Any], custom_objects: dict[str, Any] | None = None) -> Loss: ... def serialize(loss: KerasSerializable) -> dict[str, Any]: ...