mirror of
https://github.com/p2p-ld/numpydantic.git
synced 2024-11-10 00:34:29 +00:00
continuing to fix stub generation for 3.9
This commit is contained in:
parent
782360990e
commit
0f27c47299
1 changed files with 1 additions and 1 deletions
|
@ -24,7 +24,7 @@ def generate_ndarray_stub() -> str:
|
||||||
# Create import statements, saving aliased name of type if needed
|
# Create import statements, saving aliased name of type if needed
|
||||||
if arr.__module__.startswith("numpydantic") or arr.__module__ == "typing":
|
if arr.__module__.startswith("numpydantic") or arr.__module__ == "typing":
|
||||||
type_name = str(arr) if arr.__module__ == "typing" else arr.__name__
|
type_name = str(arr) if arr.__module__ == "typing" else arr.__name__
|
||||||
import_strings.append(f"from {arr.__module__} import {arr.__name__}")
|
import_strings.append(f"from {arr.__module__} import {type_name}")
|
||||||
else:
|
else:
|
||||||
# since other packages could use the same name for an imported object
|
# since other packages could use the same name for an imported object
|
||||||
# (eg dask and zarr both use an Array class)
|
# (eg dask and zarr both use an Array class)
|
||||||
|
|
Loading…
Reference in a new issue