Merge pull request #2175 from opentensor/fix/init-tao-flows

Re-init tao flows

Author: sam0x17

pallets/subtensor/src/coinbase/subnet_emissions.rs

@@ -76,10 +76,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

@@ -159,7 +159,9 @@ mod hooks {
                 // Migrate Kappa to default (0.5)
                 .saturating_add(migrations::migrate_kappa_map_to_default::migrate_kappa_map_to_default::<T>())
                 // Remove obsolete map entries
-                .saturating_add(migrations::migrate_remove_tao_dividends::migrate_remove_tao_dividends::<T>());
+                .saturating_add(migrations::migrate_remove_tao_dividends::migrate_remove_tao_dividends::<T>())
+                // Re-init tao flows
+                .saturating_add(migrations::migrate_init_tao_flow::migrate_init_tao_flow::<T>());
             weight
         }
 

pallets/subtensor/src/migrations/migrate_init_tao_flow.rs

@@ -0,0 +1,40 @@
+use alloc::string::String;
+
+use frame_support::{traits::Get, weights::Weight};
+
+use super::*;
+
+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.
+    let mut weight = T::DbWeight::get().reads(1);
+
+    // Check if the migration has already run
+    if HasMigrationRun::<T>::get(&migration_name) {
+        log::info!(
+            "Migration '{:?}' has already run. Skipping.",
+            String::from_utf8_lossy(&migration_name)
+        );
+        return weight;
+    }
+    log::info!(
+        "Running migration '{}'",
+        String::from_utf8_lossy(&migration_name)
+    );
+
+    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);
+    weight = weight.saturating_add(T::DbWeight::get().writes(1));
+
+    log::info!(
+        "Migration '{:?}' completed.",
+        String::from_utf8_lossy(&migration_name)
+    );
+
+    // Return the migration weight.
+    weight
+}

pallets/subtensor/src/migrations/mod.rs

@@ -20,6 +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_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: 335,
+    spec_version: 336,
     impl_version: 1,
     apis: RUNTIME_API_VERSIONS,
     transaction_version: 1,