Skip to content

Commit

Permalink
MNT: Use ndarray for all outputs, not list for some
Browse files Browse the repository at this point in the history
  • Loading branch information
scottclowe committed Jun 4, 2021
1 parent ab32a9e commit 4898354
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions fissa/core.py
Expand Up @@ -325,6 +325,7 @@ def separation_prep(self, redo=False):

# predefine data structures
raw = [[None for t in range(self.nTrials)] for c in range(nCell)]
raw = np.asarray(raw)
roi_polys = np.copy(raw)

# store results
Expand Down Expand Up @@ -407,6 +408,7 @@ def separate(self, redo_prep=False, redo_sep=False):
# predefine data structures
sep = [[None for t in range(self.nTrials)]
for c in range(self.nCell)]
sep = np.asarray(sep)
result = np.copy(sep)
mixmat = np.copy(sep)
info = np.copy(sep)
Expand Down Expand Up @@ -499,6 +501,7 @@ def calc_deltaf(self, freq, use_raw_f0=True, across_trials=True):
"""
deltaf_raw = [[None for t in range(self.nTrials)]
for c in range(self.nCell)]
deltaf_raw = np.asarray(deltaf_raw)
deltaf_result = np.copy(deltaf_raw)

# loop over cells
Expand Down

0 comments on commit 4898354

Please sign in to comment.