Skip to content

Commit aa4a23e

Browse files
committed
fix benchmarks
1 parent 746cc61 commit aa4a23e

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

pallets/admin-utils/src/benchmarking.rs

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ use frame_benchmarking::v1::account;
1111
use frame_benchmarking::v2::*;
1212
use frame_support::BoundedVec;
1313
use frame_system::RawOrigin;
14+
use pallet_subtensor::SubnetworkN;
15+
use subtensor_runtime_common::NetUid;
1416

1517
use super::*;
1618

@@ -240,14 +242,18 @@ mod benchmarks {
240242

241243
#[benchmark]
242244
fn sudo_set_min_allowed_uids() {
245+
let netuid = NetUid::from(1);
243246
pallet_subtensor::Pallet::<T>::set_admin_freeze_window(0);
244247
pallet_subtensor::Pallet::<T>::init_new_network(
245-
1u16.into(), /*netuid*/
248+
netuid,
246249
1u16, /*tempo*/
247250
);
251+
252+
// Artificially set that some neurons are already registered
253+
SubnetworkN::<T>::set(netuid, 32);
248254

249255
#[extrinsic_call]
250-
_(RawOrigin::Root, 1u16.into()/*netuid*/, 32u16/*max_allowed_uids*/)/*sudo_set_max_allowed_uids*/;
256+
_(RawOrigin::Root, netuid, 16u16/*min_allowed_uids*/)/*sudo_set_min_allowed_uids*/;
251257
}
252258

253259
#[benchmark]
@@ -453,7 +459,7 @@ mod benchmarks {
453459
);
454460

455461
#[extrinsic_call]
456-
_(RawOrigin::Root, 1u16.into()/*netuid*/, 4097u16/*max_allowed_uids*/)/*sudo_trim_to_max_allowed_uids()*/;
462+
_(RawOrigin::Root, 1u16.into()/*netuid*/, 256u16/*max_n*/)/*sudo_trim_to_max_allowed_uids()*/;
457463
}
458464

459465
//impl_benchmark_test_suite!(AdminUtils, crate::mock::new_test_ext(), crate::mock::Test);

0 commit comments

Comments
 (0)