zypchn commited on
Commit
ebd335b
·
verified ·
1 Parent(s): 9919e1b

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +3 -3
tools.py CHANGED
@@ -5,9 +5,9 @@ from rapidfuzz import process, fuzz
5
  try:
6
  knowledge_base = pd.read_csv("data/knowledge_base.csv")
7
  knowledge_base_latest = pd.read_csv("data/knowledge_base_latest.csv")
8
- ALL_PROD_NAMES = knowledge_base_latest['prod_name'].unique()
9
- ALL_ARTIST_NAMES = knowledge_base_latest['artist'].unique()
10
- ALL_SET_NAMES = knowledge_base_latest['set_name'].unique()
11
 
12
  except Exception as e:
13
  # Handle data loading error
 
5
  try:
6
  knowledge_base = pd.read_csv("data/knowledge_base.csv")
7
  knowledge_base_latest = pd.read_csv("data/knowledge_base_latest.csv")
8
+ ALL_PROD_NAMES = knowledge_base.prod_name.values
9
+ ALL_ARTIST_NAMES = knowledge_base.artist.values
10
+ ALL_SET_NAMES = knowledge_base.set_name.values
11
 
12
  except Exception as e:
13
  # Handle data loading error