[Fix] Small fixes to account for changes to `RegistryMixin`

neuralmagic / deepsparse

Sparsity-aware deep learning inference runtime for CPUs

https://neuralmagic.com/deepsparse/

Other

2.94k stars 169 forks source link

[Fix] Small fixes to account for changes to `RegistryMixin` #1558

Closed dbogunowicz closed 5 months ago

dbogunowicz commented 5 months ago

In this PR: https://github.com/neuralmagic/sparsezoo/pull/404 I added minor improvements to the RegistryMixin that we need to account for (they are breaking some of the tests in the GHA). To be cherry-picked to 1.7 once it lands to main

rahul-tuli commented 5 months ago

LGTM!