Fix try-runtime (#1672)

bifrost-io · Feb 27, 2025 · 6081150 · 6081150
1 parent 4ea9bcc
commit 6081150
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/runtime/bifrost-kusama/src/lib.rs b/runtime/bifrost-kusama/src/lib.rs
@@ -1736,13 +1736,13 @@ impl FailedMigrationHandler for UnfreezeChainOnFailedMigration {
 
 impl pallet_migrations::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
-	#[cfg(not(feature = "runtime-benchmarks"))]
+	#[cfg(not(any(feature = "try-runtime", feature = "runtime-benchmarks")))]
 	type Migrations = bifrost_vesting::migrations::v2::LazyMigration<
 		Runtime,
 		weights::bifrost_vesting::BifrostWeight<Runtime>,
 	>;
 	// Benchmarks need mocked migrations to guarantee that they succeed.
-	#[cfg(feature = "runtime-benchmarks")]
+	#[cfg(any(feature = "try-runtime", feature = "runtime-benchmarks"))]
 	type Migrations = pallet_migrations::mock_helpers::MockedMigrations;
 	type CursorMaxLen = ConstU32<65_536>;
 	type IdentifierMaxLen = ConstU32<256>;

diff --git a/runtime/bifrost-polkadot/src/lib.rs b/runtime/bifrost-polkadot/src/lib.rs
@@ -1495,13 +1495,13 @@ impl FailedMigrationHandler for UnfreezeChainOnFailedMigration {
 
 impl pallet_migrations::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
-	#[cfg(not(feature = "runtime-benchmarks"))]
+	#[cfg(not(any(feature = "try-runtime", feature = "runtime-benchmarks")))]
 	type Migrations = bifrost_vesting::migrations::v2::LazyMigration<
 		Runtime,
 		weights::bifrost_vesting::BifrostWeight<Runtime>,
 	>;
 	// Benchmarks need mocked migrations to guarantee that they succeed.
-	#[cfg(feature = "runtime-benchmarks")]
+	#[cfg(any(feature = "try-runtime", feature = "runtime-benchmarks"))]
 	type Migrations = pallet_migrations::mock_helpers::MockedMigrations;
 	type CursorMaxLen = ConstU32<65_536>;
 	type IdentifierMaxLen = ConstU32<256>;