Typing and general code cleanup

b_asic/simulation.py

@@ -5,6 +5,7 @@ Contains a class for simulating the result of an SFG given a set of input values
 """
 
 from collections import defaultdict
+from numbers import Number
 from typing import (
     Callable,
     List,
@@ -20,7 +21,7 @@ import numpy as np
 
 from b_asic.operation import MutableDelayMap, ResultKey
 from b_asic.signal_flow_graph import SFG
-from b_asic.types import Num, NumRuntime
+from b_asic.types import Num
 
 ResultArrayMap = Mapping[ResultKey, Sequence[Num]]
 MutableResultArrayMap = MutableMapping[ResultKey, MutableSequence[Num]]
@@ -74,7 +75,7 @@ class Simulation:
             )
         if callable(input_provider):
             self._input_functions[index] = input_provider
-        elif isinstance(input_provider, NumRuntime):
+        elif isinstance(input_provider, Number):
             self._input_functions[index] = lambda _: input_provider
         else:
             if self._input_length is None: