Skip to content

Commit dae6172

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

1 file changed

Lines changed: 24 additions & 5 deletions

File tree

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

Lines changed: 24 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
StaticPartitionsSnap,
4343
TimeWindowPartitionsSnap,
4444
)
45-
from dagster._core.definitions.selector import JobSelector
45+
from dagster._core.definitions.selector import JobSelector, ScheduleSelector, SensorSelector
4646
from dagster._core.definitions.sensor_definition import SensorType
4747
from dagster._core.definitions.temporal_context import TemporalContext
4848
from dagster._core.errors import DagsterInvariantViolationError
@@ -936,15 +936,34 @@ def resolve_automationCondition(
936936
def resolve_targetingInstigators(self, graphene_info: ResolveInfo) -> Sequence[GrapheneSensor]:
937937
if isinstance(self._remote_node, RemoteWorkspaceAssetNode):
938938
# global nodes have saved references to their targeting instigators
939+
# Convert InstigatorHandles to selectors
939940
schedules = [
940941
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
942+
for handle in self._remote_node.get_targeting_schedule_handles()
943+
if (
944+
schedule := graphene_info.context.get_schedule(
945+
ScheduleSelector(
946+
location_name=handle.repository_handle.location_name,
947+
repository_name=handle.repository_handle.repository_name,
948+
schedule_name=handle.instigator_name,
949+
)
950+
)
951+
)
952+
is not None
943953
]
944954
sensors = [
945955
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
956+
for handle in self._remote_node.get_targeting_sensor_handles()
957+
if (
958+
sensor := graphene_info.context.get_sensor(
959+
SensorSelector(
960+
location_name=handle.repository_handle.location_name,
961+
repository_name=handle.repository_handle.repository_name,
962+
sensor_name=handle.instigator_name,
963+
)
964+
)
965+
)
966+
is not None
948967
]
949968

950969
else:

0 commit comments

Comments
 (0)