Add mock imports for RTD