mle-infrastructure / mle-toolbox

Lightweight Tool to Manage Distributed ML Experiments 🛠
https://mle-infrastructure.github.io/mle_toolbox/toolbox/
MIT License
3 stars 1 forks source link

Reduce package dependencies #2

Closed denisalevi closed 3 years ago

denisalevi commented 3 years ago

Hey Rob,

here are my changes. Its mostly moving imports around and updating the dependency list. I removed all the specified dependency versions. If there are some problems with specific versions, maybe you want to add the constraints again?

The following dependencies are now optional and should probabaly be documented in the README: torch, tensorboardX (I removed tensorboard as it is not directly used in the code, but probably necessary for the visualization?), jax, gym, scikit-optimize.

I ran a pip install in a clean conda environment and tried the ODE example. It works. I didn't try the other examples yet. And you might want to add in the README that the other examples need of course torch etc. installed. :)