Function add_pymgrid_yaml_representers

src/pymgrid/utils/serialize.py:10–31

()

def add_pymgrid_yaml_representers():
    add_numpy_pandas_representers()
    from pymgrid.microgrid.trajectory import (
        DeterministicTrajectory,
        StochasticTrajectory,
        FixedLengthStochasticTrajectory
    )

    from pymgrid.microgrid.reward_shaping import (
        PVCurtailmentShaper,
        BatteryDischargeShaper,
        RescaleShaper,
        LearnedScaleRescaleShaper,
        BaselineShaper,
        SequentialShaper
    )

    from pymgrid.modules.battery.transition_models import (
        BatteryTransitionModel,
        BiasedTransitionModel,
        DecayTransitionModel
    )


def dump_data(data_dict, stream, yaml_tag):