[Typing][A-96, B-43] Add type annotations for python/paddle/hapi/callbacks.py, python/paddle/framework/framework.py (#65777)

---------

Co-authored-by: SigureMo <[email protected]>

Author: zrr1999

python/paddle/base/layer_helper_base.py

@@ -14,6 +14,7 @@
 from __future__ import annotations
 
 import copy
+from typing import TYPE_CHECKING
 
 import numpy as np
 
@@ -33,12 +34,15 @@
 from .initializer import _global_bias_initializer, _global_weight_initializer
 from .param_attr import ParamAttr, WeightNormParamAttr
 
+if TYPE_CHECKING:
+    from paddle._typing.dtype_like import _DTypeLiteral
+
 __all__ = []
 
 
 class LayerHelperBase:
     # global dtype
-    __dtype = "float32"
+    __dtype: _DTypeLiteral = "float32"
 
     def __init__(self, name, layer_type):
         self._layer_type = layer_type
@@ -265,9 +269,11 @@ def __weight_normalize(g, v, dim):
             # to achieve the subset.
             w = paddle.tensor.math._multiply_with_axis(
                 x=v,
-                y=scale
-                if dim is None
-                else paddle.reshape(x=scale, shape=[v.shape[dim]]),
+                y=(
+                    scale
+                    if dim is None
+                    else paddle.reshape(x=scale, shape=[v.shape[dim]])
+                ),
                 axis=-1 if dim is None else dim,
             )
             # To serialize the original parameter for inference, maybe a

python/paddle/framework/framework.py

@@ -12,19 +12,23 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 
 import numpy as np
 
 import paddle
 from paddle.base.data_feeder import convert_dtype
-
-# TODO: define framework api
 from paddle.base.layer_helper_base import LayerHelperBase
 
+if TYPE_CHECKING:
+    from paddle._typing.dtype_like import DTypeLike, _DTypeLiteral
+
 __all__ = []
 
 
-def set_default_dtype(d):
+def set_default_dtype(d: DTypeLike) -> None:
     """
     Set default dtype. The default dtype is initially float32.
 
@@ -73,14 +77,14 @@ def set_default_dtype(d):
     LayerHelperBase.set_default_dtype(d)
 
 
-def get_default_dtype():
+def get_default_dtype() -> _DTypeLiteral:
     """
     Get the current default dtype. The default dtype is initially float32.
 
     Args:
         None.
     Returns:
-        String, this global dtype only supports float16, float32, float64.
+        str, this global dtype only supports float16, float32, float64.
 
     Examples:
         .. code-block:: python