File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -301,7 +301,7 @@ def process_weights_for_netuid(
301
301
final_weights_count = (
302
302
torch .tensor (list (range (len (final_weights ))))
303
303
if use_torch ()
304
- else np .arrange (len (final_weights ))
304
+ else np .arange (len (final_weights ))
305
305
)
306
306
return (
307
307
(final_weights_count , final_weights )
@@ -322,12 +322,12 @@ def process_weights_for_netuid(
322
322
weights [non_zero_weight_idx ] += non_zero_weights
323
323
logging .debug (f"final_weights: { weights } " )
324
324
normalized_weights = normalize_max_weight (x = weights , limit = max_weight_limit )
325
- nw_arrange = (
325
+ nw_arange = (
326
326
torch .tensor (list (range (len (normalized_weights ))))
327
327
if use_torch ()
328
- else np .arrange (len (normalized_weights ))
328
+ else np .arange (len (normalized_weights ))
329
329
)
330
- return nw_arrange , normalized_weights
330
+ return nw_arange , normalized_weights
331
331
332
332
logging .debug (f"non_zero_weights: { non_zero_weights } " )
333
333
You can’t perform that action at this time.
0 commit comments