microsoft / hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.
MIT License
3.32k stars 274 forks source link

Need new test dataset for xgb #660

Closed ksaur closed 1 year ago

ksaur commented 1 year ago

Build workflow run

==================================== ERRORS ====================================
_______________ ERROR collecting tests/test_xgboost_converter.py _______________
ImportError while importing test module '/home/runner/work/hummingbird/hummingbird/tests/test_xgboost_converter.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/opt/hostedtoolcache/Python/3.9.15/x64/lib/python3.9/importlib/__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
tests/test_xgboost_converter.py:8: in <module>
    from sklearn.datasets import load_boston
/opt/hostedtoolcache/Python/3.9.15/x64/lib/python3.9/site-packages/sklearn/datasets/__init__.py:156: in __getattr__
    raise ImportError(msg)
E   ImportError: 
E   `load_boston` has been removed from scikit-learn since version 1.2.