Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Chris Brozdowski <[email protected]>
  • Loading branch information
edeno and CBroz1 authored Feb 21, 2024
1 parent f92715f commit 3e15a55
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/spyglass/position/v1/dlc_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
from tqdm import tqdm as tqdm

from spyglass.common.common_behav import VideoFile
from spyglass.utils import logger
from spyglass.settings import dlc_output_dir, dlc_video_dir, raw_dir


Expand Down Expand Up @@ -420,8 +421,7 @@ def get_video_path(key):
"""
import pynwb

valid_fields = VideoFile.fetch().dtype.fields.keys()
vf_key = {k: val for k, val in key.items() if k in valid_fields}
vf_key = {k: val for k, val in key.items() if k in VideoFile.heading.names}
if not VideoFile & vf_key:
VideoFile()._no_transaction_make(vf_key, verbose=False)
video_query = VideoFile & vf_key
Expand Down

0 comments on commit 3e15a55

Please sign in to comment.