Pacify mypy

pytorch · Jan 22, 2021 · e8b121e · e8b121e
1 parent c3f6710
commit e8b121e
Showing 1 changed file with 1 addition and 1 deletion.
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):