Ошибка фрейма данных pandas в pytorch lightning neptune

Question

Регистратор neptune хорошо работает в pytorch lightning, пока я не попытаюсь загрузить pandas Dataframe.

Код теста:

      data = {'col_1': [3, 2, 1, 0], 'col_2': ['a', 'b', 'c', 'd']}
data_pd = pd.DataFrame.from_dict(data)
data_html = File.as_html(data_pd)
self.logger.experiment['val/report'].upload(data_html)

Ошибка:

          self.logger.experiment['val/report'].upload(data_html)
TypeError: list indices must be integers or slices, not st

Кто-нибудь знает как это решить? Большое спасибо.