Skip to content

Commit 82ecfd9

Browse files
committed
switch from selectors => handles off of the remote node
1 parent 820a75b commit 82ecfd9

1 file changed

Lines changed: 28 additions & 5 deletions

File tree

python_modules/dagster-graphql/dagster_graphql/schema/asset_graph.py

Lines changed: 28 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,11 @@
4949
from dagster._core.event_api import AssetRecordsFilter
5050
from dagster._core.events import DagsterEventType
5151
from dagster._core.remote_representation.external import RemoteJob, RemoteRepository, RemoteSensor
52-
from dagster._core.remote_representation.external_data import AssetNodeSnap
52+
from dagster._core.remote_representation.external_data import (
53+
AssetNodeSnap,
54+
ScheduleSelector,
55+
SensorSelector,
56+
)
5357
from dagster._core.snap.node import GraphDefSnap, OpDefSnap
5458
from dagster._core.storage.asset_check_execution_record import AssetCheckInstanceSupport
5559
from dagster._core.storage.event_log.base import AssetRecord
@@ -936,15 +940,34 @@ def resolve_automationCondition(
936940
def resolve_targetingInstigators(self, graphene_info: ResolveInfo) -> Sequence[GrapheneSensor]:
937941
if isinstance(self._remote_node, RemoteWorkspaceAssetNode):
938942
# global nodes have saved references to their targeting instigators
943+
# Convert InstigatorHandles to selectors
939944
schedules = [
940945
schedule
941-
for schedule_selector in self._remote_node.get_targeting_schedule_selectors()
942-
if (schedule := graphene_info.context.get_schedule(schedule_selector)) is not None
946+
for handle in self._remote_node.get_targeting_schedule_handles()
947+
if (
948+
schedule := graphene_info.context.get_schedule(
949+
ScheduleSelector(
950+
location_name=handle.repository_handle.location_name,
951+
repository_name=handle.repository_handle.repository_name,
952+
schedule_name=handle.instigator_name,
953+
)
954+
)
955+
)
956+
is not None
943957
]
944958
sensors = [
945959
sensor
946-
for sensor_selector in self._remote_node.get_targeting_sensor_selectors()
947-
if (sensor := graphene_info.context.get_sensor(sensor_selector)) is not None
960+
for handle in self._remote_node.get_targeting_sensor_handles()
961+
if (
962+
sensor := graphene_info.context.get_sensor(
963+
SensorSelector(
964+
location_name=handle.repository_handle.location_name,
965+
repository_name=handle.repository_handle.repository_name,
966+
sensor_name=handle.instigator_name,
967+
)
968+
)
969+
)
970+
is not None
948971
]
949972

950973
else:

0 commit comments

Comments
 (0)