A test script to fine tune Gemma 2 model on python dataset