Made get_dataset_features use feature vector