Update torch_geometric/transforms/random_rotate.py

Co-authored-by: Jinu Sunil <jinu.sunil@gmail.com>
pyg-team · Oct 15, 2022 · 2465288 · 2465288
1 parent 8aa67fb
commit 2465288
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/transforms/random_rotate.py b/torch_geometric/transforms/random_rotate.py
@@ -29,7 +29,7 @@ def __init__(self, degrees: Union[Tuple[float, float], float], axis: int=0):
         self.degrees = degrees
         self.axis = axis
 
-    def __call__(self, data) -> Data:
+    def __call__(self, data: Data) -> Data:
         degree = math.pi * random.uniform(*self.degrees) / 180.0
         sin, cos = math.sin(degree), math.cos(degree)