@@ -11,6 +11,8 @@ use frame_benchmarking::v1::account;
1111use frame_benchmarking:: v2:: * ;
1212use frame_support:: BoundedVec ;
1313use frame_system:: RawOrigin ;
14+ use pallet_subtensor:: SubnetworkN ;
15+ use subtensor_runtime_common:: NetUid ;
1416
1517use 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