Fix inflation rewards bench

moondance-labs · Dec 31, 2024 · 9785935 · 9785935
1 parent b0f366f
commit 9785935
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pallets/inflation-rewards/src/lib.rs b/pallets/inflation-rewards/src/lib.rs
@@ -28,7 +28,8 @@ mod mock;
 #[cfg(test)]
 mod tests;
 
-use tp_traits::AuthorNotingInfo;
+#[cfg(any(test, feature = "runtime-benchmarks"))]
+use tp_traits::BlockNumber;
 use {
     dp_core::ParaId,
     frame_support::{
@@ -45,7 +46,8 @@ use {
         Perbill,
     },
     tp_traits::{
-        AuthorNotingHook, DistributeRewards, GetCurrentContainerChains, MaybeSelfChainBlockAuthor,
+        AuthorNotingHook, AuthorNotingInfo, DistributeRewards, GetCurrentContainerChains,
+        MaybeSelfChainBlockAuthor,
     },
 };