After the move of this class to sparse, move the import in the top of the file. normally we have all import there. In the AdvancedSubtensor1, there will be an exception, but I don't see the need for the others.