Fix model.predict output for use in benchmark

huawei-noah / SMARTS

Scalable Multi-Agent RL Training School for Autonomous Driving

MIT License

925 stars 186 forks source link

Closed Adaickalavan closed 1 year ago

Adaickalavan commented 1 year ago

model.predict() outputs a zero dimension array in a non-vectorized single env setup.
The above output could not be used as a dictionary index.
Hence, I cast the zero dimension array to int.