diff --git a/scripts/prompt_formatter.py b/scripts/prompt_formatter.py index 851a3a5..3da5a49 100644 --- a/scripts/prompt_formatter.py +++ b/scripts/prompt_formatter.py @@ -43,7 +43,7 @@ def format_prompt(*prompts: tuple[dict]): prompt = pipeline.align_colons(prompt) prompt = pipeline.align_commas(prompt, do_it=SPACE_COMMAS) prompt = pipeline.align_alternating(prompt) - prompt = pipeline.bracket_to_weights(prompt) + prompt = pipeline.bracket_to_weights(prompt, do_it=BRACKET2WEIGHT) ret.append(prompt.strip()) diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py index 90535c5..7255083 100644 --- a/tests/test_pipeline.py +++ b/tests/test_pipeline.py @@ -124,5 +124,7 @@ def test_space_to_underscore(): assert pipeline.space_to_underscore('one two three') == 'one_two_three' assert pipeline.space_to_underscore('this is a test') == 'this_is_a_test' assert pipeline.space_to_underscore(', baz') == ', baz' - assert pipeline.space_to_underscore('some_var_name', opposite=False) == 'some var name' + + pipeline.BRACKET2WEIGHT = False + assert pipeline.space_to_underscore('some_var_name', opposite=pipeline.BRACKET2WEIGHT) == 'some var name'