updates for equiv to work after datasets.py revisions