fix: add dict as return type of serialize.

hoel-bagard · hoel-bagard · commit d25b89f8a91a · 2024-02-29T18:20:59.000+09:00
diff --git a/stubs/tensorflow/tensorflow/keras/activations.pyi b/stubs/tensorflow/tensorflow/keras/activations.pyi
@@ -28,7 +28,7 @@ def relu(
     threshold: FloatTensorCompatible | FloatDataSequence = 0.0,
 ) -> Tensor: ...
 def selu(x: _ActivationInput) -> Tensor: ...
-def serialize(activation: Callable[..., Any], use_legacy_format: bool = False) -> str: ...
+def serialize(activation: Callable[..., Any], use_legacy_format: bool = False) -> str | dict[str, str]: ...
 def sigmoid(x: _ActivationInput) -> Tensor: ...
 def softmax(x: Tensor, axis: Integer = -1) -> Tensor: ...
 def softplus(x: _ActivationInput) -> Tensor: ...
