research-rainfallradar/aimodel/src/lib/dataset/__init__.py