Wouldn't it be cleaner to import next from theano.compat and keep 3.x block only?