e547a82c24b37c157bc9b40d2724a7b1fd0a7b0f,cesium/featurize.py,,save_featureset,#,346

Before Change

        Path to store feature data.
    
    features, channels = zip(*fset.columns)
    for v in [fset.values, features, channels] + list(kwargs.values()):
        if np.array(v).dtype == np.object:
            raise ValueError("Cannot save objects in .npz format.")
    np.savez(path, values=fset.values, index=fset.index.values.astype(str),

After Change

    kwargs["features"] = fset.T

    for k, v in kwargs.items():
        if isinstance(v, pd.DataFrame):
            arr = v.to_records()
            dt_list = arr.dtype.descr
            // Change type of indices from object to str
            for i, (name, dt) in enumerate(dt_list):
                if dt.endswith("O"):
                    size = max(len(x) for x in arr["index"])
                    dt_list[i] = (name, "U" + str(size))
            kwargs[k] = arr.astype(dt_list)

    // Bypass savez to allow for `allow_pickle` keyword
    // See https://github.com/numpy/numpy/pull/5770
    np.lib.npyio._savez(path, [], kwargs, compress=True, allow_pickle=False)


def load_featureset(path):

Instances