diff --git a/prompting/utils/config.py b/prompting/utils/config.py index de22e4062..4e5991f97 100644 --- a/prompting/utils/config.py +++ b/prompting/utils/config.py @@ -303,13 +303,6 @@ def parse_probabilities(prob_list): except ValueError: raise argparse.ArgumentTypeError("All probabilities must be floats.") - def parse_probabilities(prob_list): - try: - # Convert each item in the list to a float - return [float(p) for p in prob_list] - except ValueError: - raise argparse.ArgumentTypeError("All probabilities must be floats.") - parser.add_argument( "--neuron.task_p", type=parse_probabilities, # Use the custom parsing function