diff --git a/fieldExtraction/src/investment/code_funcs.py b/fieldExtraction/src/investment/code_funcs.py index 4a9db75..81d9b04 100644 --- a/fieldExtraction/src/investment/code_funcs.py +++ b/fieldExtraction/src/investment/code_funcs.py @@ -240,6 +240,10 @@ def get_code_description(code, code_mapping): return "N/A" elif "[" in code and "]" in code: code_list = ast.literal_eval(code) + elif "," in code: + code_list = code.split(",") + else: + code_list = [code] elif isinstance(code, list): code_list = code else: