Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

check is streamable #166

Closed
wants to merge 7 commits into from
Closed

check is streamable #166

Changes from 1 commit
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
3a24b56
check is streamable
AlexTMallen Apr 4, 2023
faa8496
rename yield_prompts in tests
AlexTMallen Apr 4, 2023
5259b26
Merge branch 'main' into check-streamable
norabelrose Apr 6, 2023
7aff456
check is streamable on-the-fly
AlexTMallen Apr 8, 2023
a4c2d27
Merge branch 'check-streamable' of github.com:EleutherAI/elk into che…
AlexTMallen Apr 8, 2023
bd16ce6
remove is_streamable from __all__
AlexTMallen Apr 9, 2023
78ecfca
Merge branch 'main' into check-streamable
AlexTMallen Apr 11, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
rename yield_prompts in tests
  • Loading branch information
@AlexTMallen
AlexTMallen committed Apr 4, 2023
commit faa8496ed88d5020b73758c0461713666921d938
4 changes: 2 additions & 2 deletions tests/test_load_prompts.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from elk.extraction import load_prompts, PromptConfig
from elk.extraction import yield_prompts, PromptConfig
from elk.promptsource.templates import DatasetTemplates
from itertools import cycle, islice
from typing import Literal
Expand All @@ -8,7 +8,7 @@
@pytest.mark.filterwarnings("ignore:Unable to find a decoding function")
def test_load_prompts():
def test_single_split(cfg: PromptConfig, split_type: Literal["train", "val"]):
prompt_ds = load_prompts(
prompt_ds = yield_prompts(
*cfg.datasets,
split_type=split_type,
)
Expand Down