diff --git a/onnxconverter_common/utils.py b/onnxconverter_common/utils.py index 0b18b4c..d71a40d 100644 --- a/onnxconverter_common/utils.py +++ b/onnxconverter_common/utils.py @@ -143,6 +143,18 @@ def h2o_installed(): return True +def hummingbird_installed(): + """ + Checks that *Hummingbird* is available. + """ + try: + import hummingbird.ml # noqa: F401 + + return True + except ImportError: + return False + + def get_producer(): """ Internal helper function to return the producer