Infer huggingface datasets features from a single example

infer_features.py

import datasets
from datasets import Value,Sequence,Features
def dict_generator(indict):
    if isinstance(indict,str):
        return datasets.Value("string")
    elif isinstance(indict,int):
        return datasets.Value("int32")
    elif isinstance(indict,float):
        return datasets.Value("float32")
    elif isinstance(indict,list):
        return datasets.Sequence(dict_generator(indict[0]))
    elif isinstance(indict,dict):
        out_dict = {}
        for key,value in indict.items():
            if key not in ["entities","answer_url"]:
                out_dict[key] = dict_generator(value)
        return out_dict
def infer_features(example):
  return Features(dict_generator(example))