Update signature of wrapped methods

src/python_spectrometer/daq/simulator.py

@@ -70,6 +70,20 @@ def with_delay(meth):
 
                 yield data
 
+    # Insert parameter sig
+    meth_sig = inspect.signature(meth)
+    wrapped_sig = inspect.signature(wrapped)
+
+    delay_param = inspect.Parameter('delay', inspect.Parameter.KEYWORD_ONLY, default=True,
+                                    annotation=bool | float)
+
+    parameters = list(meth_sig.parameters.values())
+    if parameters[-1].kind is inspect.Parameter.VAR_KEYWORD:
+        parameters = parameters[:-1] + [delay_param, parameters[-1]]
+    else:
+        parameters = parameters + [delay_param]
+
+    wrapped.__signature__ = wrapped_sig.replace(parameters=parameters)
     wrapped.__with_delay = True
     return wrapped