search

neuralmagic / sparsezoo

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes
Apache License 2.0
371 stars 25 forks source link

Fix misleading docstring in `RegistryMixin` #416

Closed rahul-tuli closed 10 months ago

rahul-tuli commented 10 months ago

This PR fixes a misleading docstring in RegistryMixin Also adds a test to check behavior

$ pytest -v tests/sparsezoo/utils/test_registry.py::TestRegistryFlowSingle::test_alias_is_standardized       
=============================================== test session starts ================================================
platform linux -- Python 3.11.7, pytest-7.4.4, pluggy-1.3.0 -- /home/rahul/venvs/.base_venv/bin/python3.11
cachedir: .pytest_cache
rootdir: /home/rahul/projects/sparsezoo
plugins: flaky-3.7.0, anyio-4.2.0, mock-3.12.0, hydra-core-1.3.2
collected 1 item                                                                                                   

tests/sparsezoo/utils/test_registry.py::TestRegistryFlowSingle::test_alias_is_standardized PASSED