Create Custom Env class according to Metagrok's design

input: poke-env's battle class

output: torch.Tensor(float32) all obs concatenated together (including those huge ass one-hot vectors) for the model to parse