search

FLAIROx / JaxMARL

Multi-Agent Reinforcement Learning with JAX
Apache License 2.0
393 stars 68 forks source link

Best way to create new SMAX scenarios? #88

Closed Chulabhaya closed 4 months ago

Chulabhaya commented 4 months ago

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?

benellis3 commented 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.

Chulabhaya commented 4 months ago

Thanks so much @benellis3 , I think this answers my question!