diff --git a/src/spyglass/utils/dj_mixin.py b/src/spyglass/utils/dj_mixin.py index bd6e0b244..62bbc9af6 100644 --- a/src/spyglass/utils/dj_mixin.py +++ b/src/spyglass/utils/dj_mixin.py @@ -61,7 +61,10 @@ def _nwb_table_tuple(self) -> tuple: Used to determine fetch_nwb behavior. Also used in Merge.fetch_nwb. Implemented as a cached_property to avoid circular imports.""" - from spyglass.common.common_nwbfile import AnalysisNwbfile, Nwbfile # noqa F401 + from spyglass.common.common_nwbfile import ( + AnalysisNwbfile, + Nwbfile, + ) # noqa F401 table_dict = { AnalysisNwbfile: "analysis_file_abs_path", @@ -71,9 +74,7 @@ def _nwb_table_tuple(self) -> tuple: resolved = getattr(self, "_nwb_table", None) or ( AnalysisNwbfile if "-> AnalysisNwbfile" in self.definition - else Nwbfile - if "-> Nwbfile" in self.definition - else None + else Nwbfile if "-> Nwbfile" in self.definition else None ) if not resolved: