diff --git a/gnes/preprocessor/helper.py b/gnes/preprocessor/helper.py index d723a1b9..2badba2f 100644 --- a/gnes/preprocessor/helper.py +++ b/gnes/preprocessor/helper.py @@ -357,7 +357,7 @@ def motion_algo(distances: List[float], **kwargs) -> List[int]: shots = [] num_frames = len(distances) + 1 - p = peakutils.indexes(np.array(distances).astype('float32'), thres=arg_dict['threshold'], min_dist=arg_dict['min_dist']) + p = peakutils.indexes(np.array(distances).astype('float32'), thres=arg_dict['threshold'], min_dist=arg_dict['min_dist']) if len(distances) else [] if len(p) == 0: return [0, num_frames]