Closed Chulabhaya closed 4 months ago
Exactly -- All you have to do is create a Scenario
, register it, and then pass it in to the scenario
argument of any SMAX class.
Strictly, you don't even have to register it you can just put it in, but we probably can't guarantee that won't change in the far future, although it's not in the plans to change it at the moment.
Thanks so much @benellis3 , I think this answers my question!
Hi all! First off this is awesome work and I'm having quite a bit of fun fiddling around with some of the algorithms, so thank you! I just have a rather simple question, which is what is the recommended way to create new SMAX scenarios? I see that the main SMAX environment file has some code that demonstrates how the existing scenarios are set up: https://github.com/FLAIROx/JaxMARL/blob/main/jaxmarl/environments/smax/smax_env.py#L43
Is setting up my own custom scenarios where I can vary the number of units/type of units as simple as following that structure but registering the environments in some external file instead of the main SMAX class?