import sys
from .core import jacve, tree_allclose
from .equinox_bindings import filter_jacve

from .perf import (measure_execution_time, 
                    measure_execution_time_with_jax,
                    plot_performance,
                    plot_performance_over_size,
                    plot_performance_jax_only)

if sys.version_info[:2] >= (3, 8):
    # TODO: Import directly (no need for conditional) when `python_requires = >= 3.8`
    from importlib.metadata import PackageNotFoundError, version  # pragma: no cover
else:
    from importlib_metadata import PackageNotFoundError, version  # pragma: no cover

try:
    # Change here if project is renamed and does not equal the package name
    dist_name = __name__
    __version__ = version(dist_name)
except PackageNotFoundError:  # pragma: no cover
    __version__ = "unknown"
finally:
    del version, PackageNotFoundError

