Upload entity DataFrame as a parquet artifact to MLflow.
(self, entity_df)
| 328 | _logger.debug("Failed to log entity_df metadata to MLflow: %s", e) |
| 329 | |
| 330 | def _log_entity_df_artifact(self, entity_df): |
| 331 | """Upload entity DataFrame as a parquet artifact to MLflow.""" |
| 332 | try: |
| 333 | if self.mlflow is not None: |
| 334 | self.mlflow.log_entity_df_artifact(entity_df) |
| 335 | except Exception as e: |
| 336 | _logger.debug("Failed to log entity_df artifact to MLflow: %s", e) |
| 337 | |
| 338 | def _init_openlineage_emitter(self) -> Optional[Any]: |
| 339 | """Initialize OpenLineage emitter if configured and enabled.""" |
no test coverage detected