|
49 | 49 | from dagster._core.event_api import AssetRecordsFilter |
50 | 50 | from dagster._core.events import DagsterEventType |
51 | 51 | 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 | +) |
53 | 57 | from dagster._core.snap.node import GraphDefSnap, OpDefSnap |
54 | 58 | from dagster._core.storage.asset_check_execution_record import AssetCheckInstanceSupport |
55 | 59 | from dagster._core.storage.event_log.base import AssetRecord |
@@ -936,15 +940,34 @@ def resolve_automationCondition( |
936 | 940 | def resolve_targetingInstigators(self, graphene_info: ResolveInfo) -> Sequence[GrapheneSensor]: |
937 | 941 | if isinstance(self._remote_node, RemoteWorkspaceAssetNode): |
938 | 942 | # global nodes have saved references to their targeting instigators |
| 943 | + # Convert InstigatorHandles to selectors |
939 | 944 | schedules = [ |
940 | 945 | 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 |
943 | 957 | ] |
944 | 958 | sensors = [ |
945 | 959 | 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 |
948 | 971 | ] |
949 | 972 |
|
950 | 973 | else: |
|
0 commit comments