Skip to content

Commit 1cf1945

Browse files
committed
Update engine.py
1 parent 0b47bc4 commit 1cf1945

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dlclivegui/temp/engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def is_tensorflow_model_dir_path(model_path: str | Path) -> bool:
2121
if not path.is_dir():
2222
return False
2323
has_cfg = (path / "pose_cfg.yaml").is_file()
24-
has_pb = any(p.suffix.lower() == ".pb" for p in path.glob("*.pb"))
24+
has_pb = any(p.is_file() and p.suffix.lower() == ".pb" for p in path.iterdir())
2525
return has_cfg and has_pb
2626

2727
@classmethod

0 commit comments

Comments
 (0)