fix CI

[teo-milea]

Description

Please include a summary of the change, the motivation, and any additional context that will help others understand your PR. If it closes one or more open issues, please tag them as described here.

Affected Dependencies

List any dependencies that are required for this change.

How has this been tested?

• Describe the tests that you ran to verify your changes.
• Provide instructions so we can reproduce.
• List any relevant details for your test configuration.

Checklist

• I have followed the Contribution Guidelines and Code of Conduct
• I have commented my code following the OpenMined Styleguide
• I have labeled this PR with the relevant Type labels
• My changes are covered by tests

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[rasswanth-s]

Hello 👋

Thanks so much for this PR — I really appreciate the work you put into it.

A quick heads-up: the repo just went through a significant refactor, so the codebase has changed quite a bit from what this PR was built on. As a result, these changes no longer apply cleanly to the new structure, and I'm going to close this PR for now.

if you're interested in contributing to the current version, kindly have a look at the README

Thanks again for contributing — and sorry for the churn!