Skip to content

Commit bd3cfaf

Browse files
committed
Filter out files from rglob results
Signed-off-by: Domenic Barbuzzi <[email protected]>
1 parent ad11b26 commit bd3cfaf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/examples/test_quantization_w4a4_fp4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def test_quantization_w4a4_fp4_example_script(
4242
assert result.returncode == 0, gen_cmd_fail_message(command, result)
4343

4444
# verify the expected directory was generated
45-
nvfp4_dirs: List[Path] = list(tmp_path.rglob("*-NVFP4"))
45+
nvfp4_dirs: List[Path] = [p for p in tmp_path.rglob("*-NVFP4") if p.is_dir()]
4646
assert (
4747
len(nvfp4_dirs)
4848
) == 1, f"did not find exactly one generated folder: {nvfp4_dirs}"

0 commit comments

Comments
 (0)