Update torch_geometric/transforms/random_rotate.py

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

Author: rusty1s

torch_geometric/transforms/random_rotate.py

@@ -22,7 +22,7 @@ class RandomRotate(BaseTransform):
             \mathrm{degrees}]`.
         axis (int, optional): The rotation axis. (default: :obj:`0`)
     """
-    def __init__(self, degrees: Union[Tuple[float, float], float], axis=0):
+    def __init__(self, degrees: Union[Tuple[float, float], float], axis: int=0):
         if isinstance(degrees, numbers.Number):
             degrees = (-abs(degrees), abs(degrees))
         assert isinstance(degrees, (tuple, list)) and len(degrees) == 2