fix ci??

anza-xyz · Dec 20, 2024 · 266b03e · 266b03e
1 parent bac313a
commit 266b03e
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/runtime/src/installed_scheduler_pool.rs b/runtime/src/installed_scheduler_pool.rs
@@ -504,13 +504,10 @@ impl BankWithScheduler {
     }
 
     pub fn unblock_block_production(&self) {
-        self.inner
-            .with_active_scheduler(|scheduler| {
-                assert_matches!(scheduler.context().mode(), SchedulingMode::BlockProduction);
-                scheduler.unblock_scheduling();
-                Ok(())
-            })
-            .unwrap();
+        if let SchedulerStatus::Active(scheduler) = &*self.inner.scheduler.read().unwrap() {
+            assert_matches!(scheduler.context().mode(), SchedulingMode::BlockProduction);
+            scheduler.unblock_scheduling();
+        }
     }
 
     #[cfg_attr(feature = "dev-context-only-utils", qualifiers(pub))]