From 5f7fff952ecff91ac328b19b8d339b585d633bd0 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 21 Feb 2024 08:39:19 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- happypose/toolbox/inference/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/happypose/toolbox/inference/utils.py b/happypose/toolbox/inference/utils.py index 86c39692..a99b51c6 100644 --- a/happypose/toolbox/inference/utils.py +++ b/happypose/toolbox/inference/utils.py @@ -88,7 +88,11 @@ def load_pose_models( force_panda3d_renderer: bool = False, renderer_kwargs: Optional[Dict] = None, models_root: Path = EXP_DIR, -) -> Tuple[torch.nn.Module, torch.nn.Module, BatchedMeshes,]: +) -> Tuple[ + torch.nn.Module, + torch.nn.Module, + BatchedMeshes, +]: coarse_run_dir = models_root / coarse_run_id coarse_cfg: TrainingConfig = load_cfg(coarse_run_dir / "config.yaml") coarse_cfg = check_update_config_pose(coarse_cfg)