run main.py

AssertionError: The obs returned by the step() method must be a dict, actual type: <class 'numpy.ndarray'>