VimPlug/typeshed
1
0
Fork 0
mirror of https://github.com/davidhalter/typeshed.git synced 2025-12-08 21:14:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Alex Waygood
2024-04-27 09:42:34 +01:00
committed by GitHub
parent 147aa2491b
commit 2db0b6e655
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
stubs/tensorflow/METADATA.toml
View File

@@ -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"]

1
stubs/tensorflow/tensorflow/keras/losses.pyi
View File

@@ -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]: ...