Skip to content

Commit 25df7dc

Browse files
committed
fix fft freq primitive to return np.array and add json file of the primitive
1 parent 3dfa175 commit 25df7dc

File tree

2 files changed

+32
-2
lines changed

2 files changed

+32
-2
lines changed
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
{
2+
"name": "sigpro.transformations.frequency.fftfreq.fft_freq",
3+
"primitive": "sigpro.transformations.frequency.fftfreq.fft_freq",
4+
"classifiers": {
5+
"type": "transformation",
6+
"subtype": "frequency"
7+
},
8+
"produce": {
9+
"args": [
10+
{
11+
"name": "amplitude_values",
12+
"type": "numpy.ndarray"
13+
},
14+
{
15+
"name": "sampling_frequency",
16+
"type": "float"
17+
}
18+
],
19+
"output": [
20+
{
21+
"name": "amplitude_values",
22+
"type": "numpy.ndarray"
23+
},
24+
{
25+
"name": "frequency_values",
26+
"type": "numpy.ndarray"
27+
}
28+
]
29+
}
30+
}

sigpro/transformations/frequency/fftfreq.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,15 +6,15 @@ def fft_freq(amplitude_values, sampling_frequency):
66
77
Args:
88
amplitude_values (np.ndarray):
9-
A numpy array with the signal values.
9+
A numpy array with the fft values.
1010
sampling_frequency (int or float):
1111
Sampling frequency value passed in Hz.
1212
Returns:
1313
tuple:
1414
* `amplitude_values (numpy.ndarray)`
1515
* `frequency_values (numpy.ndarray)`
1616
"""
17-
amplitude_values = amplitude_values
17+
amplitude_values = np.array(amplitude_values)
1818
frequency_values = np.fft.fftfreq(len(amplitude_values), 1 / sampling_frequency)
1919

2020
return amplitude_values, frequency_values

0 commit comments

Comments
 (0)