From 6bf7ac259ab059248b68caf81d046794fc8890be Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Robert?= Date: Fri, 29 Nov 2024 12:02:16 +0100 Subject: [PATCH] BUG: fix import error against Python's optimized mode --- array_api_strict/_flags.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/array_api_strict/_flags.py b/array_api_strict/_flags.py index 2863e5f..279b0e7 100644 --- a/array_api_strict/_flags.py +++ b/array_api_strict/_flags.py @@ -169,12 +169,13 @@ def set_array_api_strict_flags( set(default_extensions)) # We have to do this separately or it won't get added as the docstring -set_array_api_strict_flags.__doc__ = set_array_api_strict_flags.__doc__.format( - supported_versions=supported_versions, - default_version=default_version, - default_extensions=default_extensions, - draft_version=draft_version, -) +if set_array_api_strict_flags.__doc__ is not None: + set_array_api_strict_flags.__doc__ = set_array_api_strict_flags.__doc__.format( + supported_versions=supported_versions, + default_version=default_version, + default_extensions=default_extensions, + draft_version=draft_version, + ) def get_array_api_strict_flags(): """