From f0531ddf7a800e8d95d6bedff6d97bcce76922a1 Mon Sep 17 00:00:00 2001 From: Greg Zaitsev Date: Tue, 7 Apr 2026 10:31:10 -0400 Subject: [PATCH] Record TAO in-flow on burned neuron registration --- pallets/subtensor/src/subnets/registration.rs | 3 +++ pallets/subtensor/src/tests/registration.rs | 6 +++++- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/pallets/subtensor/src/subnets/registration.rs b/pallets/subtensor/src/subnets/registration.rs index afb09ed0eb..2b79832158 100644 --- a/pallets/subtensor/src/subnets/registration.rs +++ b/pallets/subtensor/src/subnets/registration.rs @@ -121,6 +121,9 @@ impl Pallet { RegistrationsThisBlock::::mutate(netuid, |val| val.saturating_inc()); Self::increase_rao_recycled(netuid, registration_cost.into()); + // Record TAO inflow + Self::record_tao_inflow(netuid, actual_burn_amount); + // 12) event log::debug!("NeuronRegistered( netuid:{netuid:?} uid:{neuron_uid:?} hotkey:{hotkey:?} )"); Self::deposit_event(Event::NeuronRegistered(netuid, neuron_uid, hotkey)); diff --git a/pallets/subtensor/src/tests/registration.rs b/pallets/subtensor/src/tests/registration.rs index c3cd3acb3c..01904c013c 100644 --- a/pallets/subtensor/src/tests/registration.rs +++ b/pallets/subtensor/src/tests/registration.rs @@ -46,7 +46,8 @@ fn test_registration_ok() { mock::setup_reserves(netuid, DEFAULT_RESERVE.into(), DEFAULT_RESERVE.into()); // Make burn small and stable for this test. - SubtensorModule::set_burn(netuid, 1_000u64.into()); + let burn = 1_000_u64; + SubtensorModule::set_burn(netuid, burn.into()); let hotkey = U256::from(1); let coldkey = U256::from(667); @@ -77,6 +78,9 @@ fn test_registration_ok() { SubtensorModule::get_stake_for_uid_and_subnetwork(netuid, uid), AlphaBalance::ZERO ); + + // TAO inflow recorded + assert_eq!(SubnetTaoFlow::::get(netuid), burn as i64); }); }