Never mind, I found the solution,
To set the TZ variable using python, you need to add the below lines to your python script where you are generating your rrd graphs -
import time, os
os.environ['TZ'] = "Asia/Kolkata" # timezone you want to display graphs in
time.tzset()