Skip to content

Commit

Permalink
Update iter tests
Browse files Browse the repository at this point in the history
  • Loading branch information
raphaelsty committed Aug 6, 2023
1 parent f111e83 commit 83be549
Showing 1 changed file with 0 additions and 9 deletions.
9 changes: 0 additions & 9 deletions sparsembed/utils/iter.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,15 +35,6 @@ def iter(
... ("Banana", "Banana is a yellow fruit.", 1),
... ]
>>> for queries, documents, labels in utils.iter(
... X,
... device="cpu",
... epochs=10,
... batch_size=3,
... shuffle=False
... ):
... print(queries)
>>> for queries, documents, labels in utils.iter(
... X,
... device="cpu",
Expand Down

0 comments on commit 83be549

Please sign in to comment.