From c232eeaa891f83ff880f7ebd3425857e4480ea14 Mon Sep 17 00:00:00 2001 From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com> Date: Sun, 5 Mar 2023 12:19:14 -0800 Subject: [PATCH] Fixed pytest plugin with new import hook typechecker syntax --- jaxtyping/pytest_plugin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jaxtyping/pytest_plugin.py b/jaxtyping/pytest_plugin.py index dafd6aa..1c458f1 100644 --- a/jaxtyping/pytest_plugin.py +++ b/jaxtyping/pytest_plugin.py @@ -52,4 +52,4 @@ def pytest_configure(config): ) raise RuntimeError(message.format(", ".join(already_imported_packages))) - install_import_hook(packages, typechecker.rsplit(".", 1)) + install_import_hook(packages, typechecker)