Address PR comments

tempora/schedule.py

@@ -29,7 +29,7 @@
 import bisect
 import datetime
 import numbers
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 from .utc import fromtimestamp as from_timestamp
 from .utc import now
@@ -44,7 +44,7 @@ class DelayedCommand(datetime.datetime):
  """
 
  delay: datetime.timedelta = datetime.timedelta()
- target: datetime.timedelta = datetime.timedelta()
+ target: Any # Expected type depends on the scheduler used
 
  @classmethod
  def from_datetime(cls, other) -> Self:
@@ -171,8 +171,7 @@ class Scheduler:
  def __init__(self) -> None:
  self.queue: list[DelayedCommand] = []
 
- def add(self, command) -> None:
- assert isinstance(command, DelayedCommand)
+ def add(self, command: DelayedCommand) -> None:
  bisect.insort(self.queue, command)
 
  def run_pending(self) -> None:
@@ -186,7 +185,7 @@ def run_pending(self) -> None:
  del self.queue[0]
 
  @abc.abstractmethod
- def run(self, command) -> None:
+ def run(self, command: DelayedCommand) -> None:
  """
  Run the command
  """
@@ -197,7 +196,7 @@ class InvokeScheduler(Scheduler):
  Command targets are functions to be invoked on schedule.
  """
 
- def run(self, command) -> None:
+ def run(self, command: DelayedCommand) -> None:
  command.target()
 
 
@@ -210,5 +209,5 @@ def __init__(self, dispatch) -> None:
  super().__init__()
  self.dispatch = dispatch
 
- def run(self, command) -> None:
+ def run(self, command: DelayedCommand) -> None:
  self.dispatch(command.target)