I found an issue with pytorch dependancy

[Nitesh-joestar]

according to the setup.py of mlagents, it states that it requires a torch version>=2.1.1.
but somewhere along the way. i belive the new pytorch release uses onnxscript to export and save the onnx file.

here comes the issue:
onnxscript requires onnx>=1.16.0
mlagents requires onnx==1.15.0

they therefore cannot be in the same env without conflicts
every version of onnxscript from the very beginning requires onnx>=1.16

i found that this is a pytorch error because the newest version gives an error when saving. stating that no module named onnxscript is found. whereas pytorch 2.1.1 doesnt throw this error.
can someone update the dependency or tell me if i am wrong

[maryamziaa]

Thanks for reporting. Should be fixed in #6251