Commit b0eb38ad authored by kamran haider's avatar kamran haider
Browse files

BUG: fix featurizer NamerError in benchmark_curve.py

parent 4be27425
Loading
Loading
Loading
Loading
+2 −4
Original line number Diff line number Diff line
@@ -106,10 +106,8 @@ for dataset in datasets:
        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"
        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(