Skip to content

Commit 4d59acd

Browse files
authored
Update weight_utils.py
1 parent 25259e9 commit 4d59acd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

bittensor/utils/weight_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ def process_weights_for_netuid(
301301
final_weights_count = (
302302
torch.tensor(list(range(len(final_weights))))
303303
if use_torch()
304-
else np.arrange(len(final_weights))
304+
else np.arange(len(final_weights))
305305
)
306306
return (
307307
(final_weights_count, final_weights)
@@ -322,12 +322,12 @@ def process_weights_for_netuid(
322322
weights[non_zero_weight_idx] += non_zero_weights
323323
logging.debug(f"final_weights: {weights}")
324324
normalized_weights = normalize_max_weight(x=weights, limit=max_weight_limit)
325-
nw_arrange = (
325+
nw_arange = (
326326
torch.tensor(list(range(len(normalized_weights))))
327327
if use_torch()
328-
else np.arrange(len(normalized_weights))
328+
else np.arange(len(normalized_weights))
329329
)
330-
return nw_arrange, normalized_weights
330+
return nw_arange, normalized_weights
331331

332332
logging.debug(f"non_zero_weights: {non_zero_weights}")
333333

0 commit comments

Comments
 (0)