Source code for suboptimumg.sweep.utils

import numpy as np
from numpy.typing import NDArray




def create_steps(min_val: int | float, max_val: int | float, steps: int) -> NDArray[np.float64]:
    """
    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[np.float64]
        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)]

    return np.array(var_range, dtype=np.float64)