import sys
from bonsai_common import SimulatorSession, Schema
from microsoft_bonsai_api.simulator.client import BonsaiClientConfig
from microsoft_bonsai_api.simulator.generated.models import SimulatorInterface

class SimModel(SimulatorSession):
def get_state(self) -> Schema:
"""Called to retreive the current state of the simulator. """
    pass

def get_interface(self) -> SimulatorInterface:
"""Called to retreive the simulator interface during registration. """
    pass

def halted(self) -> bool
"""
Should return weather the episode is halted, and
no further action will result in a state.
"""
    pass

def episode_start(self, config: Schema):
""" Called at the start of each episode """
    pass

def episode_step(self, action: Schema):
""" Called for each step of the episode """
    pass

    def example():
        config = BonsaiClientConfig(argv=sys.argv)
        sim = SimModel(config)
        while sim.run():
            continue