Skip to content

Commit 5edfb0a

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 04b2e6b commit 5edfb0a

File tree

2 files changed

+13
-7
lines changed

2 files changed

+13
-7
lines changed

ibl_to_nwb/updated_conversion/brainwide_map/convert_brainwide_map_processed_only_parallel.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,11 @@
1313
from dandi.organize import organize as dandi_organize
1414
from dandi.upload import upload as dandi_upload
1515
from neuroconv.tools.data_transfers import automatic_dandi_upload
16+
from nwbinspector.tools import get_s3_urls_and_dandi_paths
1617
from one.api import ONE
1718
from pynwb import NWBHDF5IO
1819
from pynwb.image import ImageSeries
1920
from tqdm import tqdm
20-
from nwbinspector.tools import get_s3_urls_and_dandi_paths
2121

2222
from ibl_to_nwb.updated_conversion.brainwide_map import BrainwideMapConverter
2323
from ibl_to_nwb.updated_conversion.brainwide_map.datainterfaces import (
@@ -48,7 +48,7 @@ def convert_and_upload_parallel_processed_only(
4848
assert len(os.environ.get("DANDI_API_KEY", "")) > 0, "Run `export DANDI_API_KEY=...`!"
4949

5050
nwbfile_path.parent.mkdir(exist_ok=True)
51-
51+
5252
# Download behavior and spike sorted data for this session
5353
session_path = base_path / "ibl_conversion" / session
5454
cache_folder = base_path / "ibl_conversion" / session / "cache"
@@ -72,7 +72,9 @@ def convert_and_upload_parallel_processed_only(
7272
for pose_estimation_file in pose_estimation_files:
7373
camera_name = pose_estimation_file.replace("alf/_ibl_", "").replace(".dlc.pqt", "")
7474
data_interfaces.append(
75-
IblPoseEstimationInterface(one=session_one, session=session, camera_name=camera_name, include_video=False)
75+
IblPoseEstimationInterface(
76+
one=session_one, session=session, camera_name=camera_name, include_video=False
77+
)
7678
)
7779

7880
pupil_tracking_files = session_one.list_datasets(eid=session, filename="*features*")
@@ -95,14 +97,16 @@ def convert_and_upload_parallel_processed_only(
9597

9698
metadata = session_converter.get_metadata()
9799
metadata["NWBFile"]["session_id"] = metadata["NWBFile"]["session_id"] + "-processed-only"
98-
100+
99101
session_converter.run_conversion(
100102
nwbfile_path=nwbfile_path,
101103
metadata=metadata,
102104
overwrite=True,
103105
)
104106
automatic_dandi_upload(
105-
dandiset_id="000409", nwb_folder_path=nwbfile_path.parent, cleanup=cleanup, #files_mode=files_mode
107+
dandiset_id="000409",
108+
nwb_folder_path=nwbfile_path.parent,
109+
cleanup=cleanup, # files_mode=files_mode
106110
)
107111
if cleanup:
108112
rmtree(cache_folder)
@@ -152,7 +156,7 @@ def convert_and_upload_parallel_processed_only(
152156
progress_position=1 + progress_position,
153157
# stub_test=True,
154158
# files_mode="copy", # useful when debugging
155-
#cleanup=True, # causing shutil error ATM
159+
# cleanup=True, # causing shutil error ATM
156160
)
157161
)
158162
for future in as_completed(futures):

ibl_to_nwb/updated_conversion/datainterfaces/iblposeestimationinterface.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,9 @@ def add_to_nwbfile(self, nwbfile, metadata: dict):
7777
source_software="DeepLabCut",
7878
nodes=body_parts,
7979
)
80-
if self.include_video and self.one.list_datasets(eid=self.session, filename=f"raw_video_data/*{self.camera_name}*"):
80+
if self.include_video and self.one.list_datasets(
81+
eid=self.session, filename=f"raw_video_data/*{self.camera_name}*"
82+
):
8183
original_video_file = self.one.load_dataset(
8284
id=self.session, dataset=f"raw_video_data/*{self.camera_name}*", download_only=True
8385
)

0 commit comments

Comments
 (0)