From 84fa966643aa002b0075f3b4a0eddd22d1076a86 Mon Sep 17 00:00:00 2001
From: Jiali Duan <jialiduan@fb.com>
Date: Thu, 1 Sep 2022 12:37:02 -0700
Subject: [PATCH] Fix the typing for a Camera helper function

Summary: Fixed the typing for try_get_projection_transform.

Reviewed By: kjchalup

Differential Revision: D39211811

fbshipit-source-id: ef05c6b545831d1a9f3c754aeb02fb9776e360ed
---
 pytorch3d/renderer/cameras.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pytorch3d/renderer/cameras.py b/pytorch3d/renderer/cameras.py
index e67a5f63..24ee3981 100644
--- a/pytorch3d/renderer/cameras.py
+++ b/pytorch3d/renderer/cameras.py
@@ -6,7 +6,7 @@
 
 import math
 import warnings
-from typing import Callable, List, Optional, Sequence, Tuple, Union
+from typing import List, Optional, Sequence, Tuple, Union
 
 import numpy as np
 import torch
@@ -1851,7 +1851,7 @@ def get_screen_to_ndc_transform(
     return transform
 
 
-def try_get_projection_transform(cameras, kwargs) -> Optional[Callable]:
+def try_get_projection_transform(cameras: CamerasBase, kwargs) -> Optional[Transform3d]:
     """
     Try block to get projection transform.