|
17 | 17 | EventLogRecord, |
18 | 18 | EventRecordsFilter, |
19 | 19 | EventRecordsResult, |
| 20 | + PartitionKeyFilter, |
20 | 21 | RunShardedEventsCursor, |
21 | 22 | RunStatusChangeRecordsFilter, |
22 | 23 | ) |
|
27 | 28 | from dagster._core.storage.asset_check_execution_record import ( |
28 | 29 | AssetCheckExecutionRecord, |
29 | 30 | AssetCheckExecutionRecordStatus, |
| 31 | + AssetCheckPartitionInfo, |
30 | 32 | ) |
31 | 33 | from dagster._core.storage.dagster_run import DagsterRunStatsSnapshot |
32 | 34 | from dagster._core.storage.event_log.base import ( |
@@ -378,6 +380,7 @@ def _asset_check_execution_record_from_graphql(data: dict, key: AssetCheckKey): |
378 | 380 | status=AssetCheckExecutionRecordStatus(data["status"]), |
379 | 381 | event=_event_log_entry_from_graphql(data["event"]), |
380 | 382 | create_timestamp=data["createTimestamp"], |
| 383 | + partition=data.get("partition"), |
381 | 384 | ) |
382 | 385 |
|
383 | 386 |
|
@@ -1232,14 +1235,25 @@ def get_asset_check_execution_history( |
1232 | 1235 | limit: int, |
1233 | 1236 | cursor: Optional[int] = None, |
1234 | 1237 | status: Optional[AbstractSet[AssetCheckExecutionRecordStatus]] = None, |
| 1238 | + partition_filter: Optional[PartitionKeyFilter] = None, |
1235 | 1239 | ) -> Sequence[AssetCheckExecutionRecord]: |
1236 | 1240 | raise NotImplementedError("Not callable from user cloud") |
1237 | 1241 |
|
1238 | 1242 | def get_latest_asset_check_execution_by_key( |
1239 | | - self, check_keys: Sequence[AssetCheckKey] |
| 1243 | + self, |
| 1244 | + check_keys: Sequence[AssetCheckKey], |
| 1245 | + partition_filter: Optional[PartitionKeyFilter] = None, |
1240 | 1246 | ) -> Mapping[AssetCheckKey, AssetCheckExecutionRecord]: |
1241 | 1247 | raise NotImplementedError("Not callable from user cloud") |
1242 | 1248 |
|
| 1249 | + def get_asset_check_partition_info( |
| 1250 | + self, |
| 1251 | + keys: Sequence[AssetCheckKey], |
| 1252 | + after_storage_id: Optional[int] = None, |
| 1253 | + partition_keys: Optional[Sequence[str]] = None, |
| 1254 | + ) -> Sequence[AssetCheckPartitionInfo]: |
| 1255 | + raise NotImplementedError("Not callable from user cloud") |
| 1256 | + |
1243 | 1257 | def fetch_materializations( |
1244 | 1258 | self, |
1245 | 1259 | records_filter: Union[AssetKey, AssetRecordsFilter], |
|
0 commit comments