add: set default interval and warm up steps for checking

Author: Essoz

traincheck/collect_trace.py

@@ -390,7 +390,7 @@ def main():
     parser.add_argument(
         "--warm-up-steps",
         type=int,
-        default=0,
+        default=None,
         help="Number of initial steps to always instrument.",
     )
 
@@ -413,6 +413,13 @@ def main():
                 value = os.path.join(path_prefix, value)
             setattr(args, key, value)
 
+    # Set conditional defaults based on invariants presence
+    if args.invariants:
+        if args.sampling_interval is None:
+            args.sampling_interval = config.DEFAULT_CHECKING_POLICY["interval"]
+        if args.warm_up_steps is None:
+            args.warm_up_steps = config.DEFAULT_CHECKING_POLICY["warm_up"]
+
     # set up logging
     if args.debug_mode:
         logging.basicConfig(level=logging.DEBUG)

traincheck/config/config.py

@@ -100,6 +100,11 @@
     "warm_up": 1,  # default to 1 to ensure the first step is always instrumented: before warm-up is depleted, we do instrumentation with interval=1, after warm-up is depleted, we do instrumentation with the specified interval
 }
 
+DEFAULT_CHECKING_POLICY = {
+    "interval": 10,
+    "warm_up": 10,
+}
+
 DISABLE_WRAPPER = False