Commit 4be27425 authored by kamran haider's avatar kamran haider
Browse files

BUG: fix featurizer NamerError in benchmark_curve.py

parent b1383f67
Loading
Loading
Loading
Loading
+7 −0
Original line number Diff line number Diff line
@@ -104,6 +104,13 @@ for dataset in datasets:
      if pair in CheckFeaturizer:
        featurizer = CheckFeaturizer[pair][0]
        n_features = CheckFeaturizer[pair][1]
      else:
        supported_combinations = [key for key in CheckFeaturizer.keys() if pair[0] == key[0]]
        print(supported_combinations)
        supported_models = [k[1] for k in supported_combinations] 
        raise ValueError(
          "Model %s not supported for %s dataset. Please choose from the following:\n%s"
          % (pair[1], pair[0], " ".join(supported_models)))

      tasks, all_dataset, transformers = load_dataset(
          dataset, featurizer, split='index')