carlosh93 commited on
Commit
0bd89b7
·
1 Parent(s): cd010e6

fixing null issue

Browse files
Files changed (1) hide show
  1. logic/data_utils.py +5 -5
logic/data_utils.py CHANGED
@@ -221,11 +221,11 @@ class CustomHFDatasetSaver:
221
  "language": values_dic['language'] or "",
222
  "category": values_dic['category'] or "",
223
  "concept": values_dic['concept'] or "",
224
- "category_1_concepts": [""] if values_dic.get('category_1_concepts', [""])==[] else values_dic.get('category_1_concepts', [""]),
225
- "category_2_concepts": [""] if values_dic.get('category_2_concepts', [""])==[] else values_dic.get('category_2_concepts', [""]),
226
- "category_3_concepts": [""] if values_dic.get('category_3_concepts', [""])==[] else values_dic.get('category_3_concepts', [""]),
227
- "category_4_concepts": [""] if values_dic.get('category_4_concepts', [""])==[] else values_dic.get('category_4_concepts', [""]),
228
- "category_5_concepts": [""] if values_dic.get('category_5_concepts', [""])==[] else values_dic.get('category_5_concepts', [""]),
229
  "timestamp": current_timestamp,
230
  "username": values_dic['username'] or "",
231
  "password": values_dic['password'] or "",
 
221
  "language": values_dic['language'] or "",
222
  "category": values_dic['category'] or "",
223
  "concept": values_dic['concept'] or "",
224
+ "category_1_concepts": values_dic.get('category_1_concepts') or [""],
225
+ "category_2_concepts": values_dic.get('category_2_concepts') or [""],
226
+ "category_3_concepts": values_dic.get('category_3_concepts') or [""],
227
+ "category_4_concepts": values_dic.get('category_4_concepts') or [""],
228
+ "category_5_concepts": values_dic.get('category_5_concepts') or [""],
229
  "timestamp": current_timestamp,
230
  "username": values_dic['username'] or "",
231
  "password": values_dic['password'] or "",