Update scheduler.py

nadzhou · nadzhou · commit e978821ec3ee · 2025-12-05T19:09:19.000-08:00
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -97,6 +97,7 @@
 )
 from distributed.comm.addressing import addresses_from_user_args
 from distributed.compatibility import PeriodicCallback
+from distributed.condition import ConditionExtension
 from distributed.core import (
     ErrorMessage,
     OKMessage,
@@ -144,7 +145,6 @@
     scatter_to_workers,
 )
 from distributed.variable import VariableExtension
-from distributed.condition import ConditionExtension
 
 if TYPE_CHECKING:
     from typing import TypeAlias, TypeVar

Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,7 @@`
`97`	`97`	`)`
`98`	`98`	`from distributed.comm.addressing import addresses_from_user_args`
`99`	`99`	`from distributed.compatibility import PeriodicCallback`
	`100`	`+from distributed.condition import ConditionExtension`
`100`	`101`	`from distributed.core import (`
`101`	`102`	`ErrorMessage,`
`102`	`103`	`OKMessage,`
`@@ -144,7 +145,6 @@`
`144`	`145`	`scatter_to_workers,`
`145`	`146`	`)`
`146`	`147`	`from distributed.variable import VariableExtension`
`147`		`-from distributed.condition import ConditionExtension`
`148`	`148`
`149`	`149`	`if TYPE_CHECKING:`
`150`	`150`	`from typing import TypeAlias, TypeVar`