File tree Expand file tree Collapse file tree 2 files changed +6
-3
lines changed
src/guidellm/data/deserializers
tests/unit/data/deserializers Expand file tree Collapse file tree 2 files changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -244,4 +244,3 @@ def __call__(
244244 processor = processor_factory (),
245245 random_seed = random_seed ,
246246 )
247-
Original file line number Diff line number Diff line change @@ -459,7 +459,9 @@ def test_load_config_str_json(self):
459459 ### WRITTEN BY AI ###
460460 """
461461 json_str = '{"prompt_tokens": 50, "output_tokens": 25}'
462- loaded_config = config_module ._load_config_str (json_str , SyntheticTextDatasetConfig )
462+ loaded_config = config_module ._load_config_str (
463+ json_str , SyntheticTextDatasetConfig ,
464+ )
463465
464466 assert loaded_config .prompt_tokens == 50
465467 assert loaded_config .output_tokens == 25
@@ -471,7 +473,9 @@ def test_load_config_str_key_value(self):
471473 ### WRITTEN BY AI ###
472474 """
473475 kv_str = "prompt_tokens=50,output_tokens=25"
474- loaded_config = config_module ._load_config_str (kv_str , SyntheticTextDatasetConfig )
476+ loaded_config = config_module ._load_config_str (
477+ kv_str , SyntheticTextDatasetConfig ,
478+ )
475479
476480 assert loaded_config .prompt_tokens == 50
477481 assert loaded_config .output_tokens == 25
You can’t perform that action at this time.
0 commit comments