code: def __init__(
      self,
      number_of_drops: tuple[int, int],
      drop_height: tuple[int, int],
      drop_width: tuple[int, int],
  ) -> None:
      super().__init__()
      self.number_of_drops = number_of_drops
      self.drop_height = drop_height
      self.drop_width = drop_width
  
file_path: kornia/kornia/augmentation/random_generator/_2d/random_rain.py
imports: from __future__ import annotations
  import torch
  from kornia.augmentation.random_generator.base import RandomGeneratorBase, UniformDistribution
  from kornia.augmentation.utils import _adapted_rsampling, _common_param_check, _range_bound
  from kornia.core import Tensor
  from kornia.utils import _extract_device_dtype
label_desc: Use image node_id 1 to create a new RainGenerator object from inherited base classes: RandomGeneratorBase with example: obj = RainGenerator(number_of_drops, drop_height, drop_width)