IntracorticalSpellingALS/scripts_for_figures/helpers/tsdumper.py

from yaml import CDumper
from yaml.representer import SafeRepresenter
import datetime
import pandas as pd


class TSDumper(CDumper):
    pass


def timestamp_representer(dumper, data):
    return SafeRepresenter.represent_datetime(dumper, data.to_pydatetime())


TSDumper.add_representer(datetime.datetime, SafeRepresenter.represent_datetime)
TSDumper.add_representer(pd.Timestamp, timestamp_representer)