alfraser commited on
Commit
0ff95ad
·
1 Parent(s): 8b4db2f

Fixed a bug in generate data where it didn't reference the json folder in the new file sructure

Browse files
src/data_synthesis/generate_data.py CHANGED
@@ -173,7 +173,7 @@ def add_products(category: str, product_json: str, k: int) -> None:
173
 
174
 
175
  def get_categories_and_features() -> Dict[str, List[str]]:
176
- product_features_file = os.path.join(data_dir, 'product_features.json')
177
  cats_and_feats = {}
178
  with open(product_features_file, 'r') as f:
179
  feature_json = json.load(f)
 
173
 
174
 
175
  def get_categories_and_features() -> Dict[str, List[str]]:
176
+ product_features_file = os.path.join(data_dir, 'json', 'product_features.json')
177
  cats_and_feats = {}
178
  with open(product_features_file, 'r') as f:
179
  feature_json = json.load(f)