def get_df_from_artifacts_or_executions()

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

• 14 lines of code
• 5 McCabe index (conditional complexity)

  def get_df_from_artifacts_or_executions(self, objects):
    """Returns a `pd.DataFrame` of given artifacts'/executions' properties."""
    data = {}
    for obj in objects:
      col_map = {}
      if isinstance(obj, metadata_store_pb2.Artifact):
        col_map['URI'] = obj.uri
      for p in obj.properties:
        col_map[p.upper()] = _get_value_str(obj.properties[p])
      for p in obj.custom_properties:
        col_map[p.upper()] = _get_value_str(obj.custom_properties[p])
      data[obj.id] = col_map
    df = pd.DataFrame.from_dict(data=data, orient='index').fillna('-')
    df.index.name = 'ID'
    return df