Lint and test Deploy images Documentation status Python version Package version License

CMS Machine Learning Group Python package.#

The documentation of this Python package is hosted on readthedocs.

However, note that this documentation only covers the API and technical aspects of the package itself. Usage examples and further techniques for working with machine learning tools in CMS, alongside a collection of useful guidelines can be found in the general CMS ML group documentation.

Click here to submit a feature suggestion!

Table of contents#

Docker images#

To use the cmsml package via docker, checkout our DockerHub which contains tags for several Python versions.


The tests can be triggered with

pytest -n auto tests

and in general, they should be run for Python 3.7 to 3.11. To run tests in a docker container, do

# run the tests
./tests/ cmsml/cmsml

# or interactively by adding a flag "i" to the command
./tests/ cmsml/cmsml i
> python -m unittest tests

In addition, before pushing to the repository, PEP 8 compatibility should be checked with flake8 via


or using the docker container

# run the tests
./tests/ cmsml/cmsml tests/