Skip to content

Commit

Permalink
fix(tree): fix data_block_loader (#1037)
Browse files Browse the repository at this point in the history
Co-authored-by: 杭卫强 <[email protected]>
  • Loading branch information
hangweiqiang-uestc and 杭卫强 authored Aug 17, 2022
1 parent c23f763 commit 049ec1c
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions fedlearner/model/tree/trainer_master_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,7 @@ def __init__(self,
blocks = []
for filename in files:
block_id = os.path.relpath(filename, path)
fullname = os.path.join(path, filename)
block = DataBlockInfo(block_id, fullname)
block = DataBlockInfo(block_id, filename)
blocks.append(block)
self._block_map = {block.block_id: block for block in blocks}
for rnd in range(epoch_num):
Expand Down

0 comments on commit 049ec1c

Please sign in to comment.