Skip to content

Commit 01f1dc6

Browse files
Filter out files from rglob results
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 47cf19a commit 01f1dc6

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)