diff --git a/backtesting/lib.py b/backtesting/lib.py index 6164c86f..507b5529 100644 --- a/backtesting/lib.py +++ b/backtesting/lib.py @@ -17,7 +17,7 @@ from inspect import currentframe from itertools import compress from numbers import Number -from typing import Callable, Optional, Sequence, Union +from typing import Callable, Generator, Optional, Sequence, Union import numpy as np import pandas as pd @@ -332,7 +332,7 @@ def wrap_func(resampled, *args, **kwargs): def random_ohlc_data(example_data: pd.DataFrame, *, - frac=1., random_state: Optional[int] = None) -> pd.DataFrame: + frac=1., random_state: Optional[int] = None) -> Generator[pd.DataFrame, None, None]: """ OHLC data generator. The generated OHLC data has basic [descriptive statistics](https://en.wikipedia.org/wiki/Descriptive_statistics)