Skip to content

Commit

Permalink
show progress bar
Browse files Browse the repository at this point in the history
  • Loading branch information
SeanLee97 committed Jul 27, 2024
1 parent 52fe919 commit 5445616
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion angle_emb/evaluation.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def __call__(self, model: AngleBase, show_progress: bool = True, **kwargs) -> di
embeddings2 = []
for chunk in tqdm(chunked_iter(range(len(self.text1)), self.batch_size),
total=len(self.text1)//self.batch_size,
disable=show_progress):
disable=not show_progress):
batch_text1 = [self.text1[i] for i in chunk]
batch_text2 = [self.text2[i] for i in chunk]

Expand Down

0 comments on commit 5445616

Please sign in to comment.