From 50d46047f58339f69720142966a6cbff112b3c9d Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 10 Jun 2024 19:41:44 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- setup.cfg | 4 ++-- src/pybaum/equality.py | 1 + src/pybaum/tree_util.py | 1 + 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/setup.cfg b/setup.cfg index 6edd072..73bd7d1 100644 --- a/setup.cfg +++ b/setup.cfg @@ -7,7 +7,7 @@ url = https://github.com/OpenSourceEconomics/pybaum author = Janos Gabler, Tobias Raabe author_email = janos.gabler@gmail.com license = MIT -license_file = LICENSE +license_files = LICENSE platforms = unix, linux, osx, cygwin, win32 classifiers = Development Status :: 3 - Alpha @@ -22,7 +22,7 @@ classifiers = [options] packages = find: -python_requires = >=3.7 +python_requires = >=3.8 include_package_data = True package_dir = =src diff --git a/src/pybaum/equality.py b/src/pybaum/equality.py index 9c36b3c..c5795dd 100644 --- a/src/pybaum/equality.py +++ b/src/pybaum/equality.py @@ -1,4 +1,5 @@ """Functions to check equality of pytree leaves.""" + from pybaum.config import IS_JAX_INSTALLED from pybaum.config import IS_NUMPY_INSTALLED from pybaum.config import IS_PANDAS_INSTALLED diff --git a/src/pybaum/tree_util.py b/src/pybaum/tree_util.py index 769fcfe..8e79c07 100644 --- a/src/pybaum/tree_util.py +++ b/src/pybaum/tree_util.py @@ -6,6 +6,7 @@ - The treedef containing information to unflatten pytrees is implemented differently. """ + from pybaum.equality import EQUALITY_CHECKERS from pybaum.registry import get_registry from pybaum.typecheck import get_type