reverted import (microsoft#1062)

JoshTrim · Mar 26, 2024 · e69f9b8 · e69f9b8
1 parent 99a2a11
commit e69f9b8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/autogen/code_utils.py b/autogen/code_utils.py
@@ -10,8 +10,6 @@
 from hashlib import md5
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
-from _pytest import python
-
 from autogen import oai
 
 import docker

diff --git a/autogen/coding/local_commandline_code_executor.py b/autogen/coding/local_commandline_code_executor.py
@@ -144,7 +144,7 @@ def execute_code_blocks(self, code_blocks: List[CodeBlock]) -> CommandLineCodeRe
                 f.write(code)
             file_names.append(written_file)
 
-            program = sys.executable if _cmd(lang).startswith("python") else _cmd(lang)
+            program = sys.executable if lang.startswith("python") else _cmd(lang)
             cmd = [program, str(written_file.absolute())]
 
             try: