Closed VipulMascarenhas closed 2 months ago
This PR adds the following:
get_hf_model_info
> python -m pytest -q tests/unitary/with_extras/aqua/test_* ================================================ test session starts ================================================= platform darwin -- Python 3.8.19, pytest-7.4.4, pluggy-1.0.0 rootdir: /Users/user/workspace/git/accelerated-data-science configfile: pytest.ini plugins: Faker-26.0.0, cov-5.0.0, anyio-4.2.0, xdist-3.6.1 collected 208 items tests/unitary/with_extras/aqua/test_cli.py .............. [ 6%] tests/unitary/with_extras/aqua/test_common_handler.py ... [ 8%] tests/unitary/with_extras/aqua/test_decorator.py .......... [ 12%] tests/unitary/with_extras/aqua/test_deployment.py ..................... [ 23%] tests/unitary/with_extras/aqua/test_deployment_handler.py ..s...... [ 27%] tests/unitary/with_extras/aqua/test_evaluation.py ............................. [ 41%] tests/unitary/with_extras/aqua/test_evaluation_handler.py ......... [ 45%] tests/unitary/with_extras/aqua/test_finetuning.py ....... [ 49%] tests/unitary/with_extras/aqua/test_finetuning_handler.py ..... [ 51%] tests/unitary/with_extras/aqua/test_global.py ... [ 52%] tests/unitary/with_extras/aqua/test_handlers.py ................. [ 61%] tests/unitary/with_extras/aqua/test_model.py ....................... [ 72%] tests/unitary/with_extras/aqua/test_model_handler.py .............. [ 78%] tests/unitary/with_extras/aqua/test_ui.py ............... [ 86%] tests/unitary/with_extras/aqua/test_ui_handler.py .............. [ 92%] tests/unitary/with_extras/aqua/test_ui_websocket_handler.py .... [ 94%] tests/unitary/with_extras/aqua/test_utils.py ........... [100%] ========================================== 207 passed, 1 skipped in 31.06s ===========================================
Description
This PR adds the following:
get_hf_model_info
that caches the model info from Hugging Face for 5 hours. Cache size is set as 1.Tests