Merge pull request #5028 from FederatedAI/feature-2.0.0-beta-datafram…

…e_update dataframe: fix vstack column retrieval problem
FederatedAI · Aug 2, 2023 · ad9ff3f · ad9ff3f
2 parents be96985 + 3d57d21
commit ad9ff3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/fate/arch/dataframe/ops/_dimension_scaling.py b/python/fate/arch/dataframe/ops/_dimension_scaling.py
@@ -77,8 +77,8 @@ def _align_blocks(blocks, align_fields_loc=None, full_block_migrate_set=None, ds
                     row = []
                     for _field_index in _field_indexes:
                         _src_bid, _offset = align_fields_loc[_field_index]
-                        row.append(blocks[_src_bid][lid][offset].item() if isinstance(blocks[_src_bid], torch.Tensor)
-                                   else blocks[_src_bid][lid][offset])
+                        row.append(blocks[_src_bid][lid][_offset].item() if isinstance(blocks[_src_bid], torch.Tensor)
+                                   else blocks[_src_bid][lid][_offset])
 
                     _align_block.append(row)