def get_artifact_df()

in tfx_airflow/notebooks/utils.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

  def get_artifact_df(self, artifact_id):
    """Returns a `pd.DataFrame` for an artifact with `artifact_id`.

    Args:
      artifact_id: An `int` indicating the id of an artifact in the store.

    Returns:
      A `pd.DataFrame` to display the properties of the artifact corresponding
      to `artifact_id` or None if no such artifact exists in the store.
    """
    artifacts = self.metadata_store.get_artifacts_by_id([artifact_id])
    return (
        self.get_df_from_single_artifact_or_execution(artifacts[0])
        if artifacts else None
    )