forked from SSAGESLabs/PySAGES
-
Notifications
You must be signed in to change notification settings - Fork 0
/
_compat.py
20 lines (12 loc) · 592 Bytes
/
_compat.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# SPDX-License-Identifier: MIT
# See LICENSE.md and CONTRIBUTORS.md at https://github.com/SSAGESLabs/PySAGES
# flake8: noqa F401
# pylint: disable=unused-import,relative-beyond-top-level
from importlib import import_module
from platform import python_version
from plum._version import __version_tuple__ as _plum_version_tuple
# Compatibility utils
def _version_as_tuple(ver_str):
return tuple(int(i) for i in ver_str.split(".") if i.isdigit())
_jax_version_tuple = _version_as_tuple(import_module("jaxlib").__version__)
_python_version_tuple = _version_as_tuple(python_version())