I get an error when running the code estimate.py

[RuifMaxx]

Traceback (most recent call last):
  File "api-tracing-app-management-deeprest/resource-estimation/estimate.py", line 35, in <module>
    input_size=step_size).fit_and_estimate(X, y[:, :, [idx]]))
IndexError: too many indices for array: array is 1-dimensional, but 3 were indexed

[RuifMaxx]

My python version is Python 3.8.8 and Numpy is 1.20.1

[RuifMaxx]

well, the dataset in this repo(https://github.com/IBM/api-tracing-app-management/blob/deeprest/resource-estimation/input.pkl) is too small and the lens of array (y) is zero