Pacify mypy

fritzo · fritzo · commit e8b121e382e2 · 2021-01-22T15:50:31.000-05:00
diff --git a/torch/distributions/transforms.py b/torch/distributions/transforms.py
@@ -205,7 +205,7 @@ class _InverseTransform(Transform):
     """
     def __init__(self, transform: Transform):
         super(_InverseTransform, self).__init__(cache_size=transform._cache_size)
-        self._inv = transform
+        self._inv: Transform = transform
 
     @constraints.dependent_property(is_discrete=False)
     def domain(self):
