Rename the class and add a parameter, 'dimension' to support both versions 18a7736 sionic commited on Oct 4, 2023