diff --git a/deepdoc/parser/resume/entities/corporations.py b/deepdoc/parser/resume/entities/corporations.py index 0396281dee..5035967303 100644 --- a/deepdoc/parser/resume/entities/corporations.py +++ b/deepdoc/parser/resume/entities/corporations.py @@ -29,11 +29,12 @@ GOODS = pd.read_csv( ).fillna(0) GOODS["cid"] = GOODS["cid"].astype(str) GOODS = GOODS.set_index(["cid"]) -CORP_TKS = json.load( - open(os.path.join(current_file_path, "res/corp.tks.freq.json"), "r",encoding="utf-8") -) -GOOD_CORP = json.load(open(os.path.join(current_file_path, "res/good_corp.json"), "r",encoding="utf-8")) -CORP_TAG = json.load(open(os.path.join(current_file_path, "res/corp_tag.json"), "r",encoding="utf-8")) +with open(os.path.join(current_file_path, "res/corp.tks.freq.json"), "r", encoding="utf-8") as f: + CORP_TKS = json.load(f) +with open(os.path.join(current_file_path, "res/good_corp.json"), "r", encoding="utf-8") as f: + GOOD_CORP = json.load(f) +with open(os.path.join(current_file_path, "res/corp_tag.json"), "r", encoding="utf-8") as f: + CORP_TAG = json.load(f) def baike(cid, default_v=0): diff --git a/deepdoc/parser/resume/entities/schools.py b/deepdoc/parser/resume/entities/schools.py index 4425236beb..5763ca48be 100644 --- a/deepdoc/parser/resume/entities/schools.py +++ b/deepdoc/parser/resume/entities/schools.py @@ -25,7 +25,8 @@ TBL = pd.read_csv( os.path.join(current_file_path, "res/schools.csv"), sep="\t", header=0 ).fillna("") TBL["name_en"] = TBL["name_en"].map(lambda x: x.lower().strip()) -GOOD_SCH = json.load(open(os.path.join(current_file_path, "res/good_sch.json"), "r",encoding="utf-8")) +with open(os.path.join(current_file_path, "res/good_sch.json"), "r", encoding="utf-8") as f: + GOOD_SCH = json.load(f) GOOD_SCH = set([re.sub(r"[,. &()()]+", "", c) for c in GOOD_SCH])