hotfix: asset selection

TEAMSchools · Jan 25, 2024 · e4e3ab0 · e4e3ab0
1 parent 4adb71b
commit e4e3ab0
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/src/teamster/core/powerschool/sensors.py b/src/teamster/core/powerschool/sensors.py
@@ -17,14 +17,15 @@
 
 def build_powerschool_sensor(
     name,
+    asset_selection: list[AssetsDefinition],
     asset_defs: list[AssetsDefinition],
     execution_timezone,
     minimum_interval_seconds=None,
 ):
     @sensor(
         name=name,
         minimum_interval_seconds=minimum_interval_seconds,
-        asset_selection=asset_defs,
+        asset_selection=asset_selection,
     )
     def _sensor(
         context: SensorEvaluationContext,

diff --git a/src/teamster/kippcamden/powerschool/sensors.py b/src/teamster/kippcamden/powerschool/sensors.py
@@ -1,10 +1,11 @@
 from teamster.core.powerschool.sensors import build_powerschool_sensor
 
 from .. import CODE_LOCATION, LOCAL_TIMEZONE
-from .assets import partition_assets
+from .assets import dcid_assets, partition_assets
 
 partition_sensor = build_powerschool_sensor(
     name=f"{CODE_LOCATION}_powerschool_partition_sensor",
+    asset_selection=[*dcid_assets, *partition_assets],
     asset_defs=partition_assets,
     execution_timezone=LOCAL_TIMEZONE,
     minimum_interval_seconds=(60 * 10),

diff --git a/src/teamster/kippmiami/powerschool/sensors.py b/src/teamster/kippmiami/powerschool/sensors.py
@@ -1,10 +1,11 @@
 from teamster.core.powerschool.sensors import build_powerschool_sensor
 
 from .. import CODE_LOCATION, LOCAL_TIMEZONE
-from .assets import partition_assets
+from .assets import dcid_assets, partition_assets
 
 partition_sensor = build_powerschool_sensor(
     name=f"{CODE_LOCATION}_powerschool_partition_sensor",
+    asset_selection=[*dcid_assets, *partition_assets],
     asset_defs=partition_assets,
     execution_timezone=LOCAL_TIMEZONE,
     minimum_interval_seconds=(60 * 10),

diff --git a/src/teamster/kippnewark/powerschool/sensors.py b/src/teamster/kippnewark/powerschool/sensors.py
@@ -1,10 +1,11 @@
 from teamster.core.powerschool.sensors import build_powerschool_sensor
 
 from .. import CODE_LOCATION, LOCAL_TIMEZONE
-from .assets import partition_assets
+from .assets import dcid_assets, partition_assets
 
 partition_sensor = build_powerschool_sensor(
     name=f"{CODE_LOCATION}_powerschool_partition_sensor",
+    asset_selection=[*dcid_assets, *partition_assets],
     asset_defs=partition_assets,
     execution_timezone=LOCAL_TIMEZONE,
     minimum_interval_seconds=(60 * 10),