shirwu commited on
Commit
f224ead
·
1 Parent(s): 9ed94f9

update link

Browse files
src/benchmarks/semistruct/amazon.py CHANGED
@@ -97,7 +97,7 @@ class AmazonSemiStruct(SemiStructureKB):
97
  processed_path = hf_hub_download(
98
  PROCESSED_DATASET["repo"],
99
  PROCESSED_DATASET["file"],
100
- repo_type="model",
101
  )
102
  with zipfile.ZipFile(processed_path, 'r') as zip_ref:
103
  zip_ref.extractall(self.root)
 
97
  processed_path = hf_hub_download(
98
  PROCESSED_DATASET["repo"],
99
  PROCESSED_DATASET["file"],
100
+ repo_type="dataset",
101
  )
102
  with zipfile.ZipFile(processed_path, 'r') as zip_ref:
103
  zip_ref.extractall(self.root)
src/benchmarks/semistruct/primekg.py CHANGED
@@ -49,7 +49,7 @@ class PrimeKGSemiStruct(SemiStructureKB):
49
  processed_path = hf_hub_download(
50
  PROCESSED_DATASET["repo"],
51
  PROCESSED_DATASET["file"],
52
- repo_type="model",
53
  )
54
  with zipfile.ZipFile(processed_path, 'r') as zip_ref:
55
  zip_ref.extractall(self.root)
 
49
  processed_path = hf_hub_download(
50
  PROCESSED_DATASET["repo"],
51
  PROCESSED_DATASET["file"],
52
+ repo_type="dataset",
53
  )
54
  with zipfile.ZipFile(processed_path, 'r') as zip_ref:
55
  zip_ref.extractall(self.root)