Fix stubtest for TensorFlow with latest keras release (#11838)

AlexWaygood · web-flow · commit 2db0b6e655ff · 2024-04-27T09:42:34.000+01:00
diff --git 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
@@ -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]: ...
 
