Skip to content

Commit

Permalink
Remove accidentally staged changes
Browse files Browse the repository at this point in the history
  • Loading branch information
viswavi committed Dec 1, 2023
1 parent 44e1fee commit 341d713
Showing 1 changed file with 0 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -99,12 +99,6 @@ def fit(self, X):

for doc_idx, document in tqdm(enumerate(self.documents)):
if document not in document_expansion_mapping:
assert document.split("\n")[0] in document_expansion_mapping
doc_name = document.split("\n")[0]
entity_expansions = document_expansion_mapping[doc_name]
cache_row = {"entity": doc_name, "expansion": entity_expansions}
self.cache_writer.write(cache_row)
elif document not in document_expansion_mapping:
if self.read_only:
continue
template_to_fill = self.construct_gpt3_template(doc_idx, instruction_only=self.instruction_only, demonstration_only=self.demonstration_only)
Expand Down

0 comments on commit 341d713

Please sign in to comment.