Allow multiple parameters on MockChild context

mlcommons · Jul 28, 2023 · 7b2748b · 7b2748b
1 parent c3cb787
commit 7b2748b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cli/medperf/tests/mocks/pexpect.py b/cli/medperf/tests/mocks/pexpect.py
@@ -2,10 +2,10 @@ class MockChild:
     def __init__(self, exitstatus):
         self.exitstatus = exitstatus
 
-    def __enter__(self):
+    def __enter__(self, *args, **kwargs):
         pass
 
-    def __exit__(self):
+    def __exit__(self, *args, **kwargs):
         self.close()
 
     def isalive(self):