Migration to remove SubnetEmaTaoFlow and cause it's re-initialization

Author: gztensor

pallets/subtensor/src/coinbase/subnet_emissions.rs

@@ -75,10 +75,11 @@ impl<T: Config> Pallet<T> {
             }
         } else {
             // Initialize EMA flow, set S(current_block) = min(price, ema_price) * init_factor
+            let init_factor = I64F64::saturating_from_num(1_000_000_000);
             let moving_price = I64F64::saturating_from_num(Self::get_moving_alpha_price(netuid));
             let current_price =
                 I64F64::saturating_from_num(T::SwapInterface::current_alpha_price(netuid));
-            let ema_flow = moving_price.min(current_price);
+            let ema_flow = init_factor.saturating_mul(moving_price.min(current_price));
             SubnetEmaTaoFlow::<T>::insert(netuid, (current_block, ema_flow));
             ema_flow
         }

pallets/subtensor/src/macros/hooks.rs

@@ -161,7 +161,7 @@ mod hooks {
                 // Remove obsolete map entries
                 .saturating_add(migrations::migrate_remove_tao_dividends::migrate_remove_tao_dividends::<T>())
                 // Re-init tao flows
-                .saturating_add(migrations::migrate_init_tao_flows::migrate_init_tao_flows::<T>());
+                .saturating_add(migrations::migrate_init_tao_flow::migrate_init_tao_flow::<T>());
             weight
         }
 

pallets/subtensor/src/migrations/migrate_init_tao_flow.rs

@@ -1,12 +1,10 @@
 use alloc::string::String;
 
-use frame_support::IterableStorageMap;
 use frame_support::{traits::Get, weights::Weight};
-use subtensor_runtime_common::NetUid;
 
 use super::*;
 
-pub fn migrate_init_tao_flows<T: Config>() -> Weight {
+pub fn migrate_init_tao_flow<T: Config>() -> Weight {
     let migration_name = b"migrate_init_tao_flow".to_vec();
 
     // Initialize the weight with one read operation.
@@ -25,23 +23,8 @@ pub fn migrate_init_tao_flows<T: Config>() -> Weight {
         String::from_utf8_lossy(&migration_name)
     );
 
-    let netuids: Vec<NetUid> = <NetworksAdded<T> as IterableStorageMap<NetUid, bool>>::iter()
-        .map(|(netuid, _)| netuid)
-        .collect();
-    weight = weight.saturating_add(T::DbWeight::get().reads(netuids.len() as u64));
-
-    for netuid in netuids.iter() {
-        if netuid.is_root() {
-            continue;
-        }
-        // Set SubnetEmaTaoFlow
-        SubnetEmaTaoFlow::<T>::insert(*netuid, I64F64::saturating_from_num(0));
-
-
-
-
-        weight = weight.saturating_add(T::DbWeight::get().writes(1));
-    }
+    let _ = SubnetEmaTaoFlow::<T>::clear(u32::MAX, None);
+    weight = weight.saturating_add(T::DbWeight::get().writes(1));
 
     // Mark the migration as completed
     HasMigrationRun::<T>::insert(&migration_name, true);

pallets/subtensor/src/migrations/mod.rs

@@ -20,7 +20,7 @@ pub mod migrate_fix_is_network_member;
 pub mod migrate_fix_root_subnet_tao;
 pub mod migrate_fix_root_tao_and_alpha_in;
 pub mod migrate_identities_v2;
-pub mod migrate_init_tao_flows;
+pub mod migrate_init_tao_flow;
 pub mod migrate_init_total_issuance;
 pub mod migrate_kappa_map_to_default;
 pub mod migrate_network_immunity_period;

runtime/src/lib.rs

@@ -220,7 +220,7 @@ pub const VERSION: RuntimeVersion = RuntimeVersion {
     //   `spec_version`, and `authoring_version` are the same between Wasm and native.
     // This value is set to 100 to notify Polkadot-JS App (https://polkadot.js.org/apps) to use
     //   the compatible custom types.
-    spec_version: 334,
+    spec_version: 335,
     impl_version: 1,
     apis: RUNTIME_API_VERSIONS,
     transaction_version: 1,