@@ -1085,7 +1085,7 @@ fn test_do_swap_success() {
1085
1085
let stake_amount = DefaultMinStake :: < Test > :: get ( ) * 10 ;
1086
1086
1087
1087
SubtensorModule :: create_account_if_non_existent ( & coldkey, & hotkey) ;
1088
- SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, origin_netuid, stake_amount) ;
1088
+ SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, origin_netuid, stake_amount, 0 ) ;
1089
1089
let alpha_before = SubtensorModule :: get_stake_for_hotkey_and_coldkey_on_subnet (
1090
1090
& hotkey,
1091
1091
& coldkey,
@@ -1180,7 +1180,7 @@ fn test_do_swap_insufficient_stake() {
1180
1180
let attempted_swap = stake_amount * 2 ;
1181
1181
1182
1182
SubtensorModule :: create_account_if_non_existent ( & coldkey, & hotkey) ;
1183
- SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid1, stake_amount) ;
1183
+ SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid1, stake_amount, 0 ) ;
1184
1184
1185
1185
assert_noop ! (
1186
1186
SubtensorModule :: do_swap_stake(
@@ -1209,7 +1209,7 @@ fn test_do_swap_wrong_origin() {
1209
1209
let stake_amount = 100_000 ;
1210
1210
1211
1211
SubtensorModule :: create_account_if_non_existent ( & real_coldkey, & hotkey) ;
1212
- SubtensorModule :: stake_into_subnet ( & hotkey, & real_coldkey, netuid1, stake_amount) ;
1212
+ SubtensorModule :: stake_into_subnet ( & hotkey, & real_coldkey, netuid1, stake_amount, 0 ) ;
1213
1213
1214
1214
assert_noop ! (
1215
1215
SubtensorModule :: do_swap_stake(
@@ -1237,7 +1237,7 @@ fn test_do_swap_minimum_stake_check() {
1237
1237
let swap_amount = 1 ;
1238
1238
1239
1239
SubtensorModule :: create_account_if_non_existent ( & coldkey, & hotkey) ;
1240
- SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid, total_stake) ;
1240
+ SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid, total_stake, 0 ) ;
1241
1241
1242
1242
assert_err ! (
1243
1243
SubtensorModule :: do_swap_stake(
@@ -1264,7 +1264,7 @@ fn test_do_swap_same_subnet() {
1264
1264
let stake_amount = DefaultMinStake :: < Test > :: get ( ) * 10 ;
1265
1265
1266
1266
SubtensorModule :: create_account_if_non_existent ( & coldkey, & hotkey) ;
1267
- SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid, stake_amount) ;
1267
+ SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid, stake_amount, 0 ) ;
1268
1268
1269
1269
let alpha_before =
1270
1270
SubtensorModule :: get_stake_for_hotkey_and_coldkey_on_subnet ( & hotkey, & coldkey, netuid) ;
@@ -1296,7 +1296,7 @@ fn test_do_swap_partial_stake() {
1296
1296
let total_stake = DefaultMinStake :: < Test > :: get ( ) * 10 ;
1297
1297
1298
1298
SubtensorModule :: create_account_if_non_existent ( & coldkey, & hotkey) ;
1299
- SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, origin_netuid, total_stake) ;
1299
+ SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, origin_netuid, total_stake, 0 ) ;
1300
1300
1301
1301
let swap_amount = total_stake / 2 ;
1302
1302
assert_ok ! ( SubtensorModule :: do_swap_stake(
@@ -1341,7 +1341,7 @@ fn test_do_swap_storage_updates() {
1341
1341
let stake_amount = DefaultMinStake :: < Test > :: get ( ) * 10 ;
1342
1342
1343
1343
SubtensorModule :: create_account_if_non_existent ( & coldkey, & hotkey) ;
1344
- SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, origin_netuid, stake_amount) ;
1344
+ SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, origin_netuid, stake_amount, 0 ) ;
1345
1345
1346
1346
let alpha = SubtensorModule :: get_stake_for_hotkey_and_coldkey_on_subnet (
1347
1347
& hotkey,
@@ -1390,7 +1390,7 @@ fn test_do_swap_multiple_times() {
1390
1390
let initial_stake = DefaultMinStake :: < Test > :: get ( ) * 10 ;
1391
1391
1392
1392
SubtensorModule :: create_account_if_non_existent ( & coldkey, & hotkey) ;
1393
- SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid1, initial_stake) ;
1393
+ SubtensorModule :: stake_into_subnet ( & hotkey, & coldkey, netuid1, initial_stake, 0 ) ;
1394
1394
1395
1395
for _ in 0 ..3 {
1396
1396
let alpha1 = SubtensorModule :: get_stake_for_hotkey_and_coldkey_on_subnet (
0 commit comments