From 1af5c324c02cb9060f90f0cda86f6b76a997fa43 Mon Sep 17 00:00:00 2001 From: hh_itx_win10 Date: Fri, 10 May 2024 14:28:17 -0700 Subject: [PATCH] chores: add logging --- code/aind_auto_train/curriculums/coupled_baiting_2p1.py | 2 ++ code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py | 2 ++ code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py | 2 ++ 3 files changed, 6 insertions(+) diff --git a/code/aind_auto_train/curriculums/coupled_baiting_2p1.py b/code/aind_auto_train/curriculums/coupled_baiting_2p1.py index 0f4f45d..338fb3f 100644 --- a/code/aind_auto_train/curriculums/coupled_baiting_2p1.py +++ b/code/aind_auto_train/curriculums/coupled_baiting_2p1.py @@ -17,6 +17,8 @@ Task, TrainingStage, DynamicForagingParas, AutoWaterMode, AdvancedBlockMode ) +from aind_auto_train import setup_logging +setup_logging() curriculum_name = Task.C1B1 curriculum_version = "2.1" diff --git a/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py b/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py index 54b4a17..c8a7b00 100644 --- a/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py +++ b/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py @@ -19,6 +19,8 @@ Task, TrainingStage, DynamicForagingParas, AutoWaterMode, AdvancedBlockMode ) +from aind_auto_train import setup_logging +setup_logging() # Note this could be any string, not necessarily one of the Task enums curriculum_name = "Uncoupled Baiting" diff --git a/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py b/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py index cf49b91..815c127 100644 --- a/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py +++ b/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py @@ -18,6 +18,8 @@ Task, TrainingStage, DynamicForagingParas, AutoWaterMode, AdvancedBlockMode ) +from aind_auto_train import setup_logging +setup_logging() # Note this could be any string, not necessarily one of the Task enums curriculum_name = Task.C0B0