kfickel/hokusai
1
0
Fork 0
Code Issues 1 Pull requests 1 Projects Releases Packages Wiki Activity Actions

fix: use list instead of tuple in extra_params for YAML safety

Browse source 
PyYAML serializes tuples as !!python/tuple tags which safe_load
rejects, corrupting the state file.
This commit is contained in:
Konstantin Fickel 2026-03-05 21:21:33 +01:00
parent d90db2933e
commit 47b5f899e6
Signed by: kfickel
GPG key ID: A793722F9933C1A5
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
hokusai/builder.py
View file

@ -91,9 +91,9 @@ def _collect_extra_params(target_name: str, config: ProjectConfig) -> dict[str,
if target_cfg.height is not None:
params["height"] = target_cfg.height
if target_cfg.reference_images:
params["reference_images"] = tuple(target_cfg.reference_images)
params["reference_images"] = list(target_cfg.reference_images)
if target_cfg.control_images:
params["control_images"] = tuple(target_cfg.control_images)
params["control_images"] = list(target_cfg.control_images)
return params

2
tests/test_builder.py
View file

@ -152,7 +152,7 @@ class TestCollectHelpers:
params = _collect_extra_params("out.png", config)
assert params["width"] == 512
assert params["height"] == 256
assert params["reference_images"] == ("ref.png",)
assert params["reference_images"] == ["ref.png"]
def test_collect_extra_params_empty(self, write_config: WriteConfig) -> None:
config = write_config({"targets": {"out.txt": {"prompt": "x"}}})