AnnaMats/ppo-Pyramids-Training

1from setuptools import setup2from mlagents.plugins import ML_AGENTS_STATS_WRITER3 4setup(5    name="mlagents_plugin_examples",6    version="0.0.1",7    # Example of how to add your own registration functions that will be called8    # by mlagents-learn.9    #10    # Here, the get_example_stats_writer() function in mlagents_plugin_examples/example_stats_writer.py11    # will get registered with the ML_AGENTS_STATS_WRITER plugin interface.12    entry_points={13        ML_AGENTS_STATS_WRITER: [14            "example=mlagents_plugin_examples.example_stats_writer:get_example_stats_writer"15        ]16    },17)18