From 4a3e0d18cbc9e00b78085cbdc5f60d36e951404b Mon Sep 17 00:00:00 2001
From: Artur Schneider <artur.schneider@epfl.ch>
Date: Thu, 20 Aug 2026 14:29:55 +0200
Subject: [PATCH] Populate base.Base table in the populate step

base.Base is a fully-implemented Computed table but nothing ever
called Base.populate() on it, so it was always empty. Wire it in the
same way as its structural twin vr4mice.Collab, via populate_pending,
in both run.py's "populate" mode and cron_scenario.py's nightly step
sequence. This unblocks np_sync.BarcodeSync, whose key_source joins
through base.Base to match VR datasets with Neuropixels recordings.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
---
 dj_pipeline/cron_scenario.py | 7 +++++++
 dj_pipeline/run.py           | 3 ++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/dj_pipeline/cron_scenario.py b/dj_pipeline/cron_scenario.py
index a46dde35..d831bff2 100755
--- a/dj_pipeline/cron_scenario.py
+++ b/dj_pipeline/cron_scenario.py
@@ -86,6 +86,7 @@ def import_core_schemas():
         from vr4mice.schema import (
             vr4mice,
             barcodes,
+            base,
             base_analysis,
             dlc,
             interpolated_trajectories,
@@ -97,6 +98,7 @@ def import_core_schemas():
         return (
             vr4mice,
             barcodes,
+            base,
             base_analysis,
             dlc,
             interpolated_trajectories,
@@ -115,6 +117,7 @@ def import_core_schemas():
         (
             vr4mice,
             barcodes,
+            base,
             base_analysis,
             dlc,
             interpolated_trajectories,
@@ -140,6 +143,10 @@ def import_core_schemas():
             "vr4mice.Collab.populate",
             lambda: populate_pending(vr4mice.Collab, vr4mice.Dataset, logger=logger),
         )
+        run_step(
+            "base.Base.populate",
+            lambda: populate_pending(base.Base, vr4mice.Dataset, logger=logger),
+        )
         run_step(
             "base_analysis.DataFrame.populate",
             lambda: populate_pending(
diff --git a/dj_pipeline/run.py b/dj_pipeline/run.py
index ff6f5214..2044b693 100644
--- a/dj_pipeline/run.py
+++ b/dj_pipeline/run.py
@@ -98,7 +98,7 @@ def check_folder_existence(folder_path):
 
     elif args.mode == "populate":
         from vr4mice.actions.populate_rig import populate_rig
-        from vr4mice.schema import vr4mice
+        from vr4mice.schema import base, vr4mice
         from vr4mice.utils.populate_helpers import populate_pending
 
         if args.aws:
@@ -114,6 +114,7 @@ def check_folder_existence(folder_path):
         # populate run.
         populate_rig(path=path, move=move)
         populate_pending(vr4mice.Collab, vr4mice.Dataset, logger=logger)
+        populate_pending(base.Base, vr4mice.Dataset, logger=logger)
         vr4mice.Batch.insert(vr4mice.Batch.contents, skip_duplicates=True)
         vr4mice.DatasetBatch().populate()