fix rate limit

Author: l0r1s

pallets/admin-utils/src/lib.rs

@@ -1924,15 +1924,15 @@ pub mod pallet {
             let maybe_owner = pallet_subtensor::Pallet::<T>::ensure_sn_owner_or_root_with_limits(
                 origin.clone(),
                 netuid,
-                &[TransactionType::SetMaxAllowedUids],
+                &[TransactionType::MaxUidsTrimming],
             )?;
 
             pallet_subtensor::Pallet::<T>::trim_to_max_allowed_uids(netuid, max_n)?;
 
             pallet_subtensor::Pallet::<T>::record_owner_rl(
                 maybe_owner,
                 netuid,
-                &[TransactionType::SetMaxAllowedUids],
+                &[TransactionType::MaxUidsTrimming],
             );
             Ok(())
         }
@@ -1948,7 +1948,8 @@ pub mod pallet {
             netuid: NetUid,
             min_allowed_uids: u16,
         ) -> DispatchResult {
-            ensure_root(origin)?;
+            pallet_subtensor::Pallet::<T>::ensure_root_with_rate_limit(origin, netuid)?;
+
             ensure!(
                 pallet_subtensor::Pallet::<T>::if_subnet_exist(netuid),
                 Error::<T>::SubnetDoesNotExist

pallets/subtensor/src/lib.rs

@@ -619,7 +619,7 @@ pub mod pallet {
     }
     #[pallet::type_value]
     /// -- Rate limit for set max allowed UIDs
-    pub fn SetMaxAllowedUidsRateLimit<T: Config>() -> u64 {
+    pub fn MaxUidsTrimmingRateLimit<T: Config>() -> u64 {
         prod_or_fast!(30 * 7200, 1)
     }
     #[pallet::type_value]

pallets/subtensor/src/utils/rate_limiting.rs

@@ -15,7 +15,7 @@ pub enum TransactionType {
     OwnerHyperparamUpdate(Hyperparameter),
     SubsubnetCountUpdate,
     SubsubnetEmission,
-    SetMaxAllowedUids,
+    MaxUidsTrimming,
 }
 
 impl TransactionType {
@@ -27,7 +27,7 @@ impl TransactionType {
             Self::RegisterNetwork => NetworkRateLimit::<T>::get(),
             Self::SubsubnetCountUpdate => SubsubnetCountSetRateLimit::<T>::get(),
             Self::SubsubnetEmission => SubsubnetEmissionRateLimit::<T>::get(),
-
+            Self::MaxUidsTrimming => MaxUidsTrimmingRateLimit::<T>::get(),
             Self::Unknown => 0, // Default to no limit for unknown types (no limit)
             _ => 0,
         }
@@ -140,7 +140,7 @@ impl From<TransactionType> for u16 {
             TransactionType::OwnerHyperparamUpdate(_) => 6,
             TransactionType::SubsubnetCountUpdate => 7,
             TransactionType::SubsubnetEmission => 8,
-            TransactionType::SetMaxAllowedUids => 9,
+            TransactionType::MaxUidsTrimming => 9,
         }
     }
 }
@@ -157,7 +157,7 @@ impl From<u16> for TransactionType {
             6 => TransactionType::OwnerHyperparamUpdate(Hyperparameter::Unknown),
             7 => TransactionType::SubsubnetCountUpdate,
             8 => TransactionType::SubsubnetEmission,
-            9 => TransactionType::SetMaxAllowedUids,
+            9 => TransactionType::MaxUidsTrimming,
             _ => TransactionType::Unknown,
         }
     }