Skip to content

Commit 166ccfc

Browse files
authored
Merge pull request #18 from ShaneIsley/claude/rlm-testing-framework-byfLB
Claude/rlm testing framework byf lb
2 parents 1f72735 + 9b7972b commit 166ccfc

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

‎rlm/environments/subprocess_repl.py‎

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -259,6 +259,7 @@ def __init__(
259259
allowed_packages: list[str] | None = None,
260260
auto_approve_packages: bool = False,
261261
package_approval_callback: Callable[[str], bool] | None = None,
262+
verbose: bool = False,
262263
**kwargs,
263264
):
264265
"""
@@ -276,6 +277,7 @@ def __init__(
276277
allowed_packages: Pre-approved packages (no prompt needed).
277278
auto_approve_packages: If True, install packages without prompting.
278279
package_approval_callback: Custom function for package approval.
280+
verbose: If True, print overhead summary on cleanup (default: False).
279281
"""
280282
super().__init__(persistent=persistent, depth=depth, **kwargs)
281283

@@ -295,6 +297,7 @@ def __init__(
295297
self.sandbox = sandbox
296298
self.auto_approve = auto_approve_packages
297299
self.approval_callback = package_approval_callback or self._default_approval
300+
self.verbose = verbose
298301

299302
# Pre-approved packages (stdlib + user-specified)
300303
self.allowed_packages: set[str] = {
@@ -869,8 +872,8 @@ def cleanup(self):
869872
pass
870873
self._socket_server = None
871874

872-
# Print overhead summary
873-
if hasattr(self, "_overhead_stats"):
875+
# Print overhead summary (only if verbose)
876+
if hasattr(self, "_overhead_stats") and getattr(self, "verbose", True):
874877
self.print_overhead_summary()
875878

876879
# Unregister from emergency cleanup

0 commit comments

Comments
 (0)