From 39fcf7a5ba281b6c1398876d1c01cdbea50b553e Mon Sep 17 00:00:00 2001 From: Eric Denovellis Date: Sat, 1 Jul 2023 15:13:16 -0700 Subject: [PATCH] Minor formatting --- loren_frank_data_processing/multiunit.py | 3 ++- loren_frank_data_processing/saving.py | 3 +-- loren_frank_data_processing/tetrodes.py | 3 ++- loren_frank_data_processing/visualization.py | 1 + 4 files changed, 6 insertions(+), 4 deletions(-) diff --git a/loren_frank_data_processing/multiunit.py b/loren_frank_data_processing/multiunit.py index e37dbe8..b9ab318 100644 --- a/loren_frank_data_processing/multiunit.py +++ b/loren_frank_data_processing/multiunit.py @@ -4,9 +4,10 @@ import numpy as np import pandas as pd import xarray as xr +from scipy.io import loadmat + from loren_frank_data_processing.core import get_data_filename, logger from loren_frank_data_processing.tetrodes import get_trial_time -from scipy.io import loadmat def get_multiunit_dataframe(tetrode_key, animals): diff --git a/loren_frank_data_processing/saving.py b/loren_frank_data_processing/saving.py index 2d004c5..d9b96a0 100644 --- a/loren_frank_data_processing/saving.py +++ b/loren_frank_data_processing/saving.py @@ -2,8 +2,7 @@ def get_analysis_file_path(processed_data_dir, animal, day, epoch): - """File path for analysis file. - """ + """File path for analysis file.""" filename = "{animal}_{day:02d}_{epoch:02d}.nc".format( animal=animal, day=day, epoch=epoch ) diff --git a/loren_frank_data_processing/tetrodes.py b/loren_frank_data_processing/tetrodes.py index 02f1844..498d73f 100644 --- a/loren_frank_data_processing/tetrodes.py +++ b/loren_frank_data_processing/tetrodes.py @@ -2,9 +2,10 @@ import numpy as np import pandas as pd -from loren_frank_data_processing.core import logger, reconstruct_time from scipy.io import loadmat +from loren_frank_data_processing.core import logger, reconstruct_time + def squeeze_matcell(x): try: diff --git a/loren_frank_data_processing/visualization.py b/loren_frank_data_processing/visualization.py index f4e11ae..7af96b0 100644 --- a/loren_frank_data_processing/visualization.py +++ b/loren_frank_data_processing/visualization.py @@ -1,6 +1,7 @@ import matplotlib.animation as animation import matplotlib.pyplot as plt import numpy as np + from loren_frank_data_processing.track_segment_classification import ( get_track_segments_from_graph, plot_track,