Handle 404 errors in ml client lib, minor fixes in SparkMLDriver