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)