numpy_typing_compat_cross_version_type_hints_quickstart.py

1import numpy as np
2from numpy_typing_compat import NDArray, ArrayLike
3
4def scale_array(data: ArrayLike, factor: float) -> NDArray[np.float64]:
5    """
6    Example function using numpy-typing-compat for cross-version type hinting.
7    """
8    arr = np.asanyarray(data, dtype=np.float64)
9    return arr * factor
10
11# Example usage
12input_data = [1.0, 2.0, 3.0]
13result = scale_array(input_data, 2.5)
14print(result)