Skip to content

Instantly share code, notes, and snippets.

@nicain

nicain/behavior_only.py

Created September 30, 2019 20:32
Show Gist options
  • Save nicain/6f255b79879cae67865a3f38404d5cfa to your computer and use it in GitHub Desktop.
Save nicain/6f255b79879cae67865a3f38404d5cfa to your computer and use it in GitHub Desktop.
Download ZIP
behavior only NWB file dev
Raw
behavior_only.py
from allensdk.internal.api.mtrain_api import MtrainApi
from allensdk.internal.api.behavior_lims_api import BehaviorLimsApi
from allensdk.brain_observatory.behavior.behavior_session import BehaviorSession
from allensdk.brain_observatory.behavior.behavior_api.behavior_nwb_api import BehaviorNwbApi
bl_api = BehaviorLimsApi()
m_api = MtrainApi()
# Get a session:
labtracks_id_list = m_api.get_subjects()
foraging_id = m_api.get_behavior_training_df(LabTracks_ID=463810)['behavior_session_uuid'].values[0]
behavior_session_id = bl_api.foraging_id_to_behavior_session_id(foraging_id)
session = BehaviorSession.from_lims(behavior_session_id)
# Write a file:
BehaviorNwbApi('tmp.nwb').save(session)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment