Skip to content

Commit 9c673b2

Browse files
committed
str to json!
Signed-off-by: Derek Kozikowski <[email protected]>
1 parent ebab63f commit 9c673b2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/examples/test_quantization_w4a4_fp4.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11

2+
import json
23
import shlex
34
from pathlib import Path
45
from typing import List
@@ -51,6 +52,6 @@ def test_quantization_w4a4_fp4_example_script(
5152
print(list(nvfp4_dirs[0].iterdir()))
5253

5354
# is the generated content in the expected format?
54-
config_json = Path(nvfp4_dirs[0] / "config.json").read_text()
55+
config_json = json.loads(Path(nvfp4_dirs[0] / "config.json").read_text())
5556
config_format = config_json["quantization_config"]["format"]
5657
assert(config_format == "nvfp4-pack-quantized")

0 commit comments

Comments
 (0)