Source code for suboptimumg.sweep.utils

from typing import Callable, Dict, Union

import numpy as np
import numpy.typing as npt




def create_steps(
    min_val: Union[int, float], max_val: Union[int, float], steps: int
) -> npt.NDArray[np.number]:
    """
    Create a list of steps between min and max (inclusive).

    Parameters
    ----------
    min_val : int or float
        Minimum value of the range
    max_val : int or float
        Maximum value of the range
    steps : int
        Number of steps between min and max

    Returns
    -------
    ndarray
        Array of values from min to max in equal steps
    """
    stepsize = (max_val - min_val) / steps
    var_range = [(min_val + i * stepsize) for i in range(steps + 1)]

    if isinstance(min_val, int) and isinstance(max_val, int):
        for i in range(steps + 1):
            var_range[i] = int(var_range[i])

    return np.array(var_range)