Update torch_geometric/transforms/random_rotate.py

Co-authored-by: Jinu Sunil <[email protected]>

Author: rusty1s

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)