fix: make sure instrumentation policy numbers are properly passed in

Author: Essoz

traincheck/instrumentor/tracer.py

@@ -543,7 +543,7 @@ def __init__(
         funcs_to_instr: Optional[list[str]] = None,
         API_dump_stack_trace: bool = False,
         sampling_interval: int = 1,
-        warm_up_steps: int = 0,
+        warm_up_steps: int = 1,
     ):
         """
         Instruments the specified target with additional tracing functionality.
@@ -605,6 +605,14 @@ def __init__(
         self.API_dump_stack_trace = API_dump_stack_trace
         self.instr_opts = config.load_instr_opts()
 
+        self.sampling_interval = sampling_interval
+        self.warm_up_steps = warm_up_steps
+        # set to config too
+        if self.sampling_interval:
+            config.INSTRUMENTATION_POLICY["interval"] = self.sampling_interval
+        if self.warm_up_steps:
+            config.INSTRUMENTATION_POLICY["warm_up"] = self.warm_up_steps
+
         if self.funcs_to_instr is not None and self.use_full_instr:
             get_instrumentation_logger_for_process().fatal(
                 "Invariants are provided but use_full_instr is True. Selective instrumentation cannot be done. Please remove the `--use-full-instr` flag or remove the invariants"