Add explicit python2 check to data_source