Merge pull request #77 from AllenNeuralDynamics/han_auto_update_curri…

…culums

feat: also update curriculums automatically

code/run_capsule.py

@@ -5,13 +5,18 @@
 
 from aind_auto_train import setup_logging, __version__
 from aind_auto_train.auto_train_manager import DynamicForagingAutoTrainManager
+from aind_auto_train.curriculum_manager import CurriculumManager, LOCAL_SAVED_CURRICULUM_ROOT
+
 setup_logging()
 logger = logging.getLogger(__name__)
 
-def update_auto_train_database(managers, interval=10):
+def update_auto_train_database(managers, curriculum_manager, interval=600):
     while True:
         logger.info(f'\n\n --- v{__version__} ---')
         try:
+            logger.info(f'-- Update curriculums --')
+            curriculum_manager.download_curriculums()
+            logger.info(f'-- Update training manager --')
             for manager_name, manager in managers.items():
                 manager.update()
         except Exception as e:
@@ -31,9 +36,15 @@ def run():
         if_rerun_all=False
     )
 
+    curriculum_manager = CurriculumManager(
+        saved_curriculums_on_s3=dict(bucket='aind-behavior-data',
+                                     root='foraging_auto_training/saved_curriculums/'),
+        saved_curriculums_local=LOCAL_SAVED_CURRICULUM_ROOT
+    )
+
     # create a thread, and run update_auto_train_database() per 1 hour
-    update_auto_train_database(managers)
-    thread = threading.Thread(update_auto_train_database, args=[managers])
+    update_auto_train_database(managers, curriculum_manager)
+    thread = threading.Thread(update_auto_train_database, args=[managers, curriculum_manager])
     thread.start()