Skip to content

Commit 9200639

Browse files
authored
fix: pin numpy to 1.x for lambda layers (#2889)
1 parent 5ee6698 commit 9200639

File tree

4 files changed

+476
-453
lines changed

4 files changed

+476
-453
lines changed

awswrangler/_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -856,7 +856,7 @@ def split_pandas_frame(df: pd.DataFrame, splits: int) -> list[pd.DataFrame]:
856856
total = len(df)
857857
each_section, extras = divmod(total, splits)
858858
section_sizes = [0] + extras * [each_section + 1] + (splits - extras) * [each_section]
859-
div_points = _nx.array(section_sizes, dtype=_nx.intp).cumsum()
859+
div_points = _nx.array(section_sizes, dtype=_nx.intp).cumsum() # type: ignore[attr-defined]
860860

861861
sub_dfs = []
862862
for i in range(splits):

awswrangler/s3/_write_dataset.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def _get_bucketing_series(df: pd.DataFrame, bucketing_info: typing.BucketingInfo
2828
axis="columns",
2929
)
3030
)
31-
return bucket_number_series.astype(np.array([pd.CategoricalDtype(range(bucketing_info[1]))]))
31+
return bucket_number_series.astype(pd.CategoricalDtype(range(bucketing_info[1])))
3232

3333

3434
def _simulate_overflow(value: int, bits: int = 31, signed: bool = False) -> int:

0 commit comments

Comments
 (0)