added the python script for KLInference