@author: anind """ from spi import SPI import numpy as np import os, random from PIL import Image import time import gdal import osr from math import isnan spi = SPI() spi.set_rolling_window_params( span=1, window_type=None, center=True) # Set distribution parameters spi.set_distribution_params(dist_type='gam') def use_spi(arr: np.array) -> np.array: data = spi.calculate(arr, starting_month=1) data = data.flatten() #Calculate and return 1d array return data