Log materialization to MLflow ops experiment.
(
self,
feature_views: List[Any],
start_date: Optional[datetime],
end_date: datetime,
duration_seconds: float,
incremental: bool = False,
)
| 2460 | raise |
| 2461 | |
| 2462 | def _mlflow_log_materialize( |
| 2463 | self, |
| 2464 | feature_views: List[Any], |
| 2465 | start_date: Optional[datetime], |
| 2466 | end_date: datetime, |
| 2467 | duration_seconds: float, |
| 2468 | incremental: bool = False, |
| 2469 | ): |
| 2470 | """Log materialization to MLflow ops experiment.""" |
| 2471 | try: |
| 2472 | if self.mlflow is None or not self.config.mlflow.log_operations: |
| 2473 | return |
| 2474 | fv_names = [getattr(fv, "name", str(fv)) for fv in feature_views] |
| 2475 | self.mlflow.log_materialize( |
| 2476 | feature_view_names=fv_names, |
| 2477 | start_date=start_date, |
| 2478 | end_date=end_date, |
| 2479 | duration_seconds=duration_seconds, |
| 2480 | incremental=incremental, |
| 2481 | ) |
| 2482 | except Exception as e: |
| 2483 | _logger.debug("MLflow materialize logging failed: %s", e) |
| 2484 | |
| 2485 | def _emit_openlineage_materialize_start( |
| 2486 | self, |
no test coverage detected