Skip to content

Commit 8095341

Browse files
varun-sundar-rabindranathVarun Sundar Rabindranath
andauthored
[misc] LoRA: Remove unused long context test data (#15558)
Signed-off-by: Varun Sundar Rabindranath <[email protected]> Co-authored-by: Varun Sundar Rabindranath <[email protected]>
1 parent 69db16a commit 8095341

File tree

3 files changed

+0
-154
lines changed

3 files changed

+0
-154
lines changed

tests/lora/conftest.py

Lines changed: 0 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -241,39 +241,6 @@ def long_context_lora_files_16k_1():
241241
return snapshot_download(repo_id="SangBinCho/long_context_16k_testing_1")
242242

243243

244-
@pytest.fixture(scope="session")
245-
def long_context_lora_files_16k_2():
246-
return snapshot_download(repo_id="SangBinCho/long_context_16k_testing_2")
247-
248-
249-
@pytest.fixture(scope="session")
250-
def long_context_lora_files_32k():
251-
return snapshot_download(repo_id="SangBinCho/long_context_32k_testing")
252-
253-
254-
@pytest.fixture(scope="session")
255-
def long_context_infos(long_context_lora_files_16k_1,
256-
long_context_lora_files_16k_2,
257-
long_context_lora_files_32k):
258-
cleanup_dist_env_and_memory(shutdown_ray=True)
259-
infos: dict[int, ContextInfo] = {}
260-
for lora_checkpoint_info in LONG_LORA_INFOS:
261-
lora_id = lora_checkpoint_info["lora_id"]
262-
if lora_id == 1:
263-
lora = long_context_lora_files_16k_1
264-
elif lora_id == 2:
265-
lora = long_context_lora_files_16k_2
266-
elif lora_id == 3:
267-
lora = long_context_lora_files_32k
268-
else:
269-
raise AssertionError("Unknown lora id")
270-
infos[lora_id] = {
271-
"context_length": lora_checkpoint_info["context_length"],
272-
"lora": lora,
273-
}
274-
return infos
275-
276-
277244
@pytest.fixture
278245
def llama_2_7b_engine_extra_embeddings():
279246
cleanup_dist_env_and_memory(shutdown_ray=True)

tests/lora/data/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)