We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0b47bc4 commit 1cf1945Copy full SHA for 1cf1945
dlclivegui/temp/engine.py
@@ -21,7 +21,7 @@ def is_tensorflow_model_dir_path(model_path: str | Path) -> bool:
21
if not path.is_dir():
22
return False
23
has_cfg = (path / "pose_cfg.yaml").is_file()
24
- has_pb = any(p.suffix.lower() == ".pb" for p in path.glob("*.pb"))
+ has_pb = any(p.is_file() and p.suffix.lower() == ".pb" for p in path.iterdir())
25
return has_cfg and has_pb
26
27
@classmethod
0 commit comments