(self, command: dict, key: str)
| 281 | assert len(name_aliases) == 1, "name_aliases list should only have one item" |
| 282 | |
| 283 | def offline_write_batch(self, command: dict, key: str): |
| 284 | self._validate_offline_write_batch_parameters(command) |
| 285 | |
| 286 | feature_view_names = command["feature_view_names"] |
| 287 | name_aliases = command["name_aliases"] |
| 288 | |
| 289 | project = self.store.config.project |
| 290 | feature_views = self.list_feature_views_by_name( |
| 291 | feature_view_names=feature_view_names, |
| 292 | name_aliases=name_aliases, |
| 293 | project=project, |
| 294 | ) |
| 295 | |
| 296 | assert len(feature_views) == 1, "incorrect feature view" |
| 297 | table = self.flights[key] |
| 298 | self.offline_store.offline_write_batch( |
| 299 | self.store.config, |
| 300 | cast( |
| 301 | FeatureView, |
| 302 | assert_permissions( |
| 303 | feature_views[0], actions=[AuthzedAction.WRITE_OFFLINE] |
| 304 | ), |
| 305 | ), |
| 306 | table, |
| 307 | command["progress"], |
| 308 | ) |
| 309 | |
| 310 | def _validate_write_logged_features_parameters(self, command: dict): |
| 311 | assert "feature_service_name" in command |
no test coverage detected