fix type hint

agimus-project · Mar 1, 2024 · 2ecce0c · 2ecce0c
1 parent 984ca89
commit 2ecce0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/happypose/pose_estimators/megapose/inference/pose_estimator.py b/happypose/pose_estimators/megapose/inference/pose_estimator.py
@@ -526,7 +526,7 @@ def run_inference_pipeline(
         bsz_objects: Optional[int] = None,
         cuda_timer: Optional[bool] = False,
         coarse_estimates: Optional[PoseEstimatesType] = None,
-        labels_to_keep: List[str] = None,
+        labels_to_keep: Optional[List[str]] = None,
     ) -> Tuple[PoseEstimatesType, dict]:
         """Runs the entire pose estimation pipeline.