Skip to content

Commit

Permalink
Fix lints
Browse files Browse the repository at this point in the history
Signed-off-by: Jonathan Shi <[email protected]>
  • Loading branch information
noloerino committed Aug 9, 2022
1 parent b10f6f5 commit 490778c
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 12 deletions.
6 changes: 3 additions & 3 deletions modin/core/dataframe/pandas/partitioning/partition_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -1234,13 +1234,13 @@ def apply_func_to_indices_both_axis(
all_parts = np.array(
[
[
partitions_copy[row_blk_idx, col_blk_idx]
for row_blk_idx, _ in row_partitions
partition_copy[row_blk_idx, col_blk_idx]
for row_blk_idx, _ in row_partitions_list
]
for col_blk_idx, _ in col_partitions_list
]
).flatten()
self._update_partition_dimension_caches(all_parts)
cls._update_partition_dimension_caches(all_parts)

def compute_part_size(indexer, remote_part, part_idx, axis):
"""Compute indexer length along the specified axis for the passed partition."""
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -72,14 +72,9 @@ def len_fn(df):
return [partition.apply(len_fn)._data]
elif partition.axis == axis:
return [
ptn.apply(len_fn)._data
for ptn in partition.list_of_block_partitions
ptn.apply(len_fn)._data for ptn in partition.list_of_block_partitions
]
return [
partition.list_of_block_partitions[0]
.apply(len_fn)
._data
]
return [partition.list_of_block_partitions[0].apply(len_fn)._data]

@property
def _row_lengths(self):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -421,7 +421,7 @@ def width(self):
for obj in self.list_of_partitions_to_combine
]
new_widths = DaskWrapper.materialize(
[cache for cache in caches if isinstance(promise, Future)]
[cache for cache in caches if isinstance(cache, Future)]
)
dask_idx = 0
for i, cache in enumerate(caches):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -427,7 +427,7 @@ def width(self):
for obj in self.list_of_partitions_to_combine
]
new_widths = ray.get(
[cache for cache in caches if isinstance(promise, ObjectIDType)]
[cache for cache in caches if isinstance(cache, ObjectIDType)]
)
ray_idx = 0
for i, cache in enumerate(caches):
Expand Down

0 comments on commit 490778c

Please sign in to comment.