p2p-ld/numpydantic
1
0
Fork 0
mirror of https://github.com/p2p-ld/numpydantic.git synced 2024-11-12 17:54:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix bad typing import generation in meta stubfile generator

Browse source
This commit is contained in:
sneakers-the-rat 2024-05-24 18:36:03 -07:00
parent 025832aa3d
commit 81006b64ec
Signed by untrusted user who does not match committer: jonny
GPG key ID: 6DCB96EF1E4D232D
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/numpydantic/meta.py
View file

@ -24,7 +24,8 @@ def generate_ndarray_stub() -> str:
# Create import statements, saving aliased name of type if needed
if arr.__module__.startswith("numpydantic") or arr.__module__ == "typing":
type_name = str(arr) if arr.__module__ == "typing" else arr.__name__
import_strings.append(f"from {arr.__module__} import {type_name}")
if arr.__module__ != "typing":
import_strings.append(f"from {arr.__module__} import {type_name}")
else:
# since other packages could use the same name for an imported object
# (eg dask and zarr both use an Array class)
@ -39,6 +40,7 @@ def generate_ndarray_stub() -> str:
type_names.append(type_name)
import_strings.extend(_BUILTIN_IMPORTS)
import_strings = list(dict.fromkeys(import_strings))
import_string = "\n".join(import_strings)
class_union = " | ".join(type_names)