diff --git a/atroposlib/envs/base.py b/atroposlib/envs/base.py index 7b4eab14..3b7f8c3c 100644 --- a/atroposlib/envs/base.py +++ b/atroposlib/envs/base.py @@ -996,8 +996,8 @@ class BaseEnv(ABC): ) # 2. OpenAI Configuration (used for potential overrides) - oai_cli_passed_args = ( - extract_namespace(cli_passed_flags, openai_full_prefix), + oai_cli_passed_args = extract_namespace( + cli_passed_flags, openai_full_prefix ) # CLI args yaml_oai_config = yaml_config.get(OPENAI_NAMESPACE, {}) if isinstance(default_server_configs, ServerBaseline) and ( @@ -1190,8 +1190,8 @@ class BaseEnv(ABC): ) # 2. OpenAI Configuration - oai_cli_passed_args = ( - extract_namespace(cli_passed_flags, openai_full_prefix), + oai_cli_passed_args = extract_namespace( + cli_passed_flags, openai_full_prefix ) # CLI args yaml_oai_config = yaml_config.get(OPENAI_NAMESPACE, {}) if isinstance(default_server_configs, ServerBaseline) and ( @@ -1218,7 +1218,7 @@ class BaseEnv(ABC): default_openai_config_.model_dump(), # Default OpenaiConfig (or from class init) PROCESS_MODE_OPENAI_DEFAULT_CONFIG.model_dump(), # Process Mode Defaults yaml_oai_config, - extract_namespace(cli_passed_flags, openai_full_prefix), + oai_cli_passed_args, ) else: openai_config_dict = {}