cms-ml / cmsml

Python Package of the CMS Machine Learning Group
https://cmsml.readthedocs.io
BSD 3-Clause "New" or "Revised" License
19 stars 6 forks source link

Add AOT tools #10

Closed riga closed 10 months ago

riga commented 10 months ago

I'm opening this PR for @Bogdan-Wiederspan who did the actual code prototyping and implementation in this PR. However, we needed access to the CI pipelines, which is why the feature branch is not part of his fork but already in the main repo.


This PR adds all tools necessary

New unit tests are provided along with docstrings for all new helpers.

Besides these additions, a few minor changes to existing code are provided, mostly to align the existing tensorflow tools to python 3.

Once merged, we should

@valsdav

valsdav commented 10 months ago

Hi @riga @Bogdan-Wiederspan , thanks a lot for the PR and for including all the tests and documetation.

I can tag the repository, but I guess you have the credentials for the update to pypi, right?

valsdav commented 10 months ago

:rocket: