flake fixes (0c6f1385) · Commits · sails-dev / sails

sails/stft.py

+29 −28

Original line number	Diff line number	Diff line
		import warnings
		import logging
		import warnings
		from dataclasses import dataclass

		import numpy as np
		@@ -269,7 +269,6 @@ def compute_multitaper_stft(x, freq_resolution=1, num_tapers='auto', time_bandwi
		logging.info('multitaper fft data {0}'.format(result.shape))
		logging.info('multitaper fft data power {0}'.format((result[0, -1, :, :]**2).sum(axis=1)))


		# Average over tapers - could be high level option? mean or median?
		result = np.average(result, weights=taper_weights, axis=-2)

		@@ -466,6 +465,8 @@ def _set_frange(fmin, fmax, fs):
		return fmin, fmax

		import typing


		@dataclass
		class STFTConfig:
		# Data specific args