Skip to content

Commit

Permalink
ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
oliche committed Dec 4, 2024
1 parent c36ead1 commit 02345f9
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 6 deletions.
4 changes: 2 additions & 2 deletions src/iblphotometry/loaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,15 +12,15 @@ def __init__(self, one, verbose=False):
self.one = one
self.verbose = verbose

def load_photometry_data(self, eid=None, pid=None, rename=True) -> nap.TsdFrame:
def load_photometry_data(self, eid=None, pid=None, rename=True) -> pd.DataFrame:
if pid is not None:
raise NotImplementedError
# return self._load_data_from_pid(pid)

if eid is not None:
return self._load_data_from_eid(eid, rename=rename)

def _load_data_from_eid(self, eid, rename=True) -> nap.TsdFrame:
def _load_data_from_eid(self, eid, rename=True) -> pd.DataFrame:
raw_photometry_df = self.one.load_dataset(eid, 'photometry.signal.pqt')
locations_df = self.one.load_dataset(eid, 'photometryROI.locations.pqt')
read_config = dict(
Expand Down
4 changes: 0 additions & 4 deletions src/iblphotometry/pipelines.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,8 @@

import numpy as np
import pandas as pd
from iblphotometry.helpers import z, filt

# from ibldsp.utils import WindowGenerator
from iblphotometry import sliding_operations
from iblphotometry import bleach_corrections
from iblphotometry.outlier_detection import remove_spikes
from iblphotometry.bleach_corrections import lowpass_bleachcorrect, isosbestic_correct
from iblphotometry.sliding_operations import sliding_mad
Expand All @@ -15,7 +12,6 @@
from iblphotometry.helpers import zscore

import logging
from copy import copy

logger = logging.getLogger()

Expand Down

0 comments on commit 02345f9

Please sign in to comment.