Skip to content

Commit 1d4a928

Browse files
committed
Speed up goofi startup
1 parent 4ce1189 commit 1d4a928

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

src/goofi/nodes/inputs/replay.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import os
2-
import pandas as pd
32
import numpy as np
43
from goofi.data import Data, DataType
54
from goofi.node import Node
@@ -28,6 +27,8 @@ def setup(self):
2827
self.load_csv()
2928

3029
def load_csv(self):
30+
import pandas as pd
31+
3132
filename = self.params["Read"]["filename"].value
3233
if filename != self.last_filename: # Only reload if filename changed
3334
if os.path.exists(filename):

src/goofi/nodes/signal/fooofaperiodic.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import numpy as np
2-
from fooof import FOOOF
32

43
from goofi.data import Data, DataType
54
from goofi.node import Node
@@ -25,12 +24,17 @@ def config_params():
2524
}
2625
}
2726

27+
def setup(self):
28+
from fooof import FOOOF
29+
30+
self.FOOOF = FOOOF
31+
2832
def process(self, psd_data: Data):
2933
if psd_data is None or psd_data.data is None:
3034
return None
3135

3236
# Create FOOOF object & set its parameters
33-
fm = FOOOF(max_n_peaks=self.params["fooof"]["max_n_peaks"].value)
37+
fm = self.FOOOF(max_n_peaks=self.params["fooof"]["max_n_peaks"].value)
3438

3539
# Extract the PSD and freqs from the Data object
3640
psd = psd_data.data

0 commit comments

Comments
 (0)