From 46f6e2cff5834c33fe5736d208bfd7e4418375b0 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 7 Dec 2023 19:52:28 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../pose_estimators/cosypose/cosypose/integrated/detector.py | 1 + happypose/pose_estimators/megapose/inference/detector.py | 1 + 2 files changed, 2 insertions(+) diff --git a/happypose/pose_estimators/cosypose/cosypose/integrated/detector.py b/happypose/pose_estimators/cosypose/cosypose/integrated/detector.py index 3a2a6d96..a968a1b1 100644 --- a/happypose/pose_estimators/cosypose/cosypose/integrated/detector.py +++ b/happypose/pose_estimators/cosypose/cosypose/integrated/detector.py @@ -12,6 +12,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") + class Detector(DetectorModule): def __init__(self, model, ds_name): super().__init__() diff --git a/happypose/pose_estimators/megapose/inference/detector.py b/happypose/pose_estimators/megapose/inference/detector.py index e15aee69..84b01dbc 100644 --- a/happypose/pose_estimators/megapose/inference/detector.py +++ b/happypose/pose_estimators/megapose/inference/detector.py @@ -31,6 +31,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") + class Detector(DetectorModule): def __init__(self, model: torch.nn.Module) -> None: super().__init__()