chores: add logging

AllenNeuralDynamics · May 10, 2024 · 1af5c32 · 1af5c32
1 parent 44dd5cb
commit 1af5c32
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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