diff --git a/tsfresh/utilities/distribution.py b/tsfresh/utilities/distribution.py index 66ac692b..f2f7ec41 100644 --- a/tsfresh/utilities/distribution.py +++ b/tsfresh/utilities/distribution.py @@ -8,6 +8,7 @@ Design of this module by Nils Braun """ +from numba import jit import itertools import math import warnings @@ -170,6 +171,7 @@ def calculate_best_chunk_size(self, data_length): chunk_size += 1 return chunk_size + @jit def map_reduce( self, map_function,