def get_df_from_single_artifact_or_execution()

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

• 10 lines of code
• 4 McCabe index (conditional complexity)

  def get_df_from_single_artifact_or_execution(self, obj):
    """Returns a `pd.DataFrame` based on an artifact/execution properties.

    Args:
      obj: An instance of `metadata_store_pb2.Artifact` or
           `metadata_store_pb2.Execution`.

    Returns:
      A `pd.DataFrame` to display the properties of an artifact/execution.
    """
    data = {}
    if isinstance(obj, metadata_store_pb2.Artifact):
      data['URI'] = obj.uri
    for p in obj.properties:
      data[p.upper()] = _get_value_str(obj.properties[p])
    for p in obj.custom_properties:
      data[p.upper()] = _get_value_str(obj.custom_properties[p])
    return pd.DataFrame.from_dict(
        data=data, orient='index', columns=['']).fillna('-')