f3207032063ae7156dc68df5c3dcd21fec0b48ca,cesium/tests/test_predict.py,,test_predict_classification,#,35

Before Change

    for model_type in classifier_types:
        model = build_model.build_model_from_featureset(fset,
                                                        model_type=model_type)
        preds = predict.predict_data_files(TS_CLASS_PATHS,
                                           list(fset.data_vars), model,
                                           custom_features_script=None)
        if preds.prediction.values.ravel()[0].dtype == np.dtype("float"):
            assert(all(preds.prediction.class_label == [b"class1", b"class2",
                                                        b"class3"]))
            assert(preds.prediction.values.shape ==

After Change

    for model_type in classifier_types:
        model = build_model.build_model_from_featureset(fset,
                                                        model_type=model_type)
        preds = prediction_task(TS_CLASS_PATHS, list(fset.data_vars), model,
                                custom_features_script=None)().get()
        if preds.prediction.values.ravel()[0].dtype == np.dtype("float"):
            assert(all(preds.prediction.class_label == [b"class1", b"class2",
                                                        b"class3"]))
            assert(preds.prediction.values.shape ==

Instances