From 4202759ace2b7bed0656e1473988bba73b11a59e Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 17 Dec 2024 13:58:55 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 ...inwide_map_processed_only_local_testing.py | 21 +++++++------------
 1 file changed, 7 insertions(+), 14 deletions(-)

diff --git a/src/ibl_to_nwb/_scripts/convert_brainwide_map_processed_only_local_testing.py b/src/ibl_to_nwb/_scripts/convert_brainwide_map_processed_only_local_testing.py
index f45f83a..4fc8da9 100644
--- a/src/ibl_to_nwb/_scripts/convert_brainwide_map_processed_only_local_testing.py
+++ b/src/ibl_to_nwb/_scripts/convert_brainwide_map_processed_only_local_testing.py
@@ -4,7 +4,6 @@
 
 import os
 from pathlib import Path
-from shutil import rmtree
 
 from one.api import ONE
 
@@ -49,7 +48,9 @@
 data_interfaces = list()
 
 # These interfaces should always be present in source data
-data_interfaces.append(IblSortingInterface(session=eid, cache_folder=one_cache_folder_path / "sorting", revision=revision))
+data_interfaces.append(
+    IblSortingInterface(session=eid, cache_folder=one_cache_folder_path / "sorting", revision=revision)
+)
 data_interfaces.append(BrainwideMapTrialsInterface(one=one, session=eid, revision=revision))
 data_interfaces.append(WheelInterface(one=one, session=eid, revision=revision))
 
@@ -57,31 +58,23 @@
 pose_estimation_files = one.list_datasets(eid=eid, filename="*.dlc*")
 for pose_estimation_file in pose_estimation_files:
     camera_name = pose_estimation_file.replace("alf/_ibl_", "").replace(".dlc.pqt", "")
-    data_interfaces.append(
-        IblPoseEstimationInterface(one=one, session=eid, camera_name=camera_name, revision=revision)
-    )
+    data_interfaces.append(IblPoseEstimationInterface(one=one, session=eid, camera_name=camera_name, revision=revision))
 
 pupil_tracking_files = one.list_datasets(eid=eid, filename="*features*")
 for pupil_tracking_file in pupil_tracking_files:
     camera_name = pupil_tracking_file.replace("alf/_ibl_", "").replace(".features.pqt", "")
-    data_interfaces.append(
-        PupilTrackingInterface(one=one, session=eid, camera_name=camera_name, revision=revision)
-    )
+    data_interfaces.append(PupilTrackingInterface(one=one, session=eid, camera_name=camera_name, revision=revision))
 
 roi_motion_energy_files = one.list_datasets(eid=eid, filename="*ROIMotionEnergy.npy*")
 for roi_motion_energy_file in roi_motion_energy_files:
     camera_name = roi_motion_energy_file.replace("alf/", "").replace(".ROIMotionEnergy.npy", "")
-    data_interfaces.append(
-        RoiMotionEnergyInterface(one=one, session=eid, camera_name=camera_name, revision=revision)
-    )
+    data_interfaces.append(RoiMotionEnergyInterface(one=one, session=eid, camera_name=camera_name, revision=revision))
 
 if one.list_datasets(eid=eid, collection="alf", filename="licks*"):
     data_interfaces.append(LickInterface(one=one, session=eid, revision=revision))
 
 # Run conversion
-session_converter = BrainwideMapConverter(
-    one=one, session=eid, data_interfaces=data_interfaces, verbose=True
-)
+session_converter = BrainwideMapConverter(one=one, session=eid, data_interfaces=data_interfaces, verbose=True)
 
 metadata = session_converter.get_metadata()
 subject_id = metadata["Subject"]["subject_id"]