Function try_cast_from_str

evals/elsuite/bugged_tools/utils.py:64–82

Given string n, cast to specified type and return. Warns and returns None if this fails

(n: str, cast_type: type)

def try_cast_from_str(n: str, cast_type: type):
    """
    Given string n, cast to specified type and return. Warns and returns None
    if this fails
    """
    if cast_type not in (str, int, float, list):
        return None

    try:
        if cast_type == str:
            return str(n)
        elif cast_type == int:
            return int(n)
        elif cast_type == float:
            return float(n)
        elif cast_type == list:
            return ast.literal_eval(n)
    except (ValueError, SyntaxError, TypeError, MemoryError, RecursionError):
        return None