As part of my B.Tech Project on modeling RL algorithms for job shop scheduling, I have written an environment which can be integrated with your RL model. Along with the environment, I have added a naive Q Learning model and Deep Q Network model for some number of machines and jobs.

Find the repository with the codes, features, and instruction for usage here

I hope this environment is found useful. Also, feel free to provide any suggestion and feedback about the same.