@@ -932,7 +932,11 @@ async def create_array(
932
932
933
933
@deprecated ("Use AsyncGroup.create_array instead." )
934
934
async def create_dataset (
935
- self , name : str , ** kwargs : Any
935
+ self ,
936
+ name : str ,
937
+ * ,
938
+ shape : ShapeLike ,
939
+ ** kwargs : Any ,
936
940
) -> AsyncArray [ArrayV2Metadata ] | AsyncArray [ArrayV3Metadata ]:
937
941
"""Create an array.
938
942
@@ -943,6 +947,8 @@ async def create_dataset(
943
947
----------
944
948
name : str
945
949
Array name.
950
+ shape : int | tuple[int, ...]
951
+ Array shape.
946
952
kwargs : dict
947
953
Additional arguments passed to :func:`zarr.AsyncGroup.create_array`.
948
954
@@ -953,7 +959,7 @@ async def create_dataset(
953
959
.. deprecated:: 3.0.0
954
960
The h5py compatibility methods will be removed in 3.1.0. Use `AsyncGroup.create_array` instead.
955
961
"""
956
- return await self .create_array (name , ** kwargs )
962
+ return await self .create_array (name , shape = shape , ** kwargs )
957
963
958
964
@deprecated ("Use AsyncGroup.require_array instead." )
959
965
async def require_dataset (
@@ -976,7 +982,7 @@ async def require_dataset(
976
982
----------
977
983
name : str
978
984
Array name.
979
- shape : int or tuple of ints
985
+ shape : int | tuple[int, ...]
980
986
Array shape.
981
987
dtype : str or dtype, optional
982
988
NumPy dtype.
@@ -1621,7 +1627,13 @@ def create_dataset(self, name: str, **kwargs: Any) -> Array:
1621
1627
return Array (self ._sync (self ._async_group .create_dataset (name , ** kwargs )))
1622
1628
1623
1629
@deprecated ("Use Group.require_array instead." )
1624
- def require_dataset (self , name : str , ** kwargs : Any ) -> Array :
1630
+ def require_dataset (
1631
+ self ,
1632
+ name : str ,
1633
+ * ,
1634
+ shape : ShapeLike ,
1635
+ ** kwargs : Any ,
1636
+ ) -> Array :
1625
1637
"""Obtain an array, creating if it doesn't exist.
1626
1638
1627
1639
Arrays are known as "datasets" in HDF5 terminology. For compatibility
@@ -1633,7 +1645,7 @@ def require_dataset(self, name: str, **kwargs: Any) -> Array:
1633
1645
----------
1634
1646
name : str
1635
1647
Array name.
1636
- shape : int or tuple of ints
1648
+ shape : int | tuple[int, ...]
1637
1649
Array shape.
1638
1650
dtype : str or dtype, optional
1639
1651
NumPy dtype.
@@ -1648,9 +1660,15 @@ def require_dataset(self, name: str, **kwargs: Any) -> Array:
1648
1660
.. deprecated:: 3.0.0
1649
1661
The h5py compatibility methods will be removed in 3.1.0. Use `Group.require_array` instead.
1650
1662
"""
1651
- return Array (self ._sync (self ._async_group .require_array (name , ** kwargs )))
1663
+ return Array (self ._sync (self ._async_group .require_array (name , shape = shape , ** kwargs )))
1652
1664
1653
- def require_array (self , name : str , ** kwargs : Any ) -> Array :
1665
+ def require_array (
1666
+ self ,
1667
+ name : str ,
1668
+ * ,
1669
+ shape : ShapeLike ,
1670
+ ** kwargs : Any ,
1671
+ ) -> Array :
1654
1672
"""Obtain an array, creating if it doesn't exist.
1655
1673
1656
1674
@@ -1660,7 +1678,7 @@ def require_array(self, name: str, **kwargs: Any) -> Array:
1660
1678
----------
1661
1679
name : str
1662
1680
Array name.
1663
- shape : int or tuple of ints
1681
+ shape : int | tuple[int, ...]
1664
1682
Array shape.
1665
1683
dtype : str or dtype, optional
1666
1684
NumPy dtype.
@@ -1672,7 +1690,7 @@ def require_array(self, name: str, **kwargs: Any) -> Array:
1672
1690
-------
1673
1691
a : Array
1674
1692
"""
1675
- return Array (self ._sync (self ._async_group .require_array (name , ** kwargs )))
1693
+ return Array (self ._sync (self ._async_group .require_array (name , shape = shape , ** kwargs )))
1676
1694
1677
1695
def empty (self , * , name : str , shape : ChunkCoords , ** kwargs : Any ) -> Array :
1678
1696
return Array (self ._sync (self ._async_group .empty (name = name , shape = shape , ** kwargs )))
0 commit comments