add name to tests

mozilla-ai · Apr 4, 2024 · c0f5a35 · c0f5a35
1 parent 1049697
commit c0f5a35
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/integration/test_finetuning.py b/tests/integration/test_finetuning.py
@@ -21,7 +21,7 @@ def job_config(llm_model_path, text_dataset_path) -> FinetuningJobConfig:
         save_steps=1,
         save_strategy="epoch",
     )
-    tracking_config = WandbRunConfig(name="test-finetuning-job")
+    tracking_config = WandbRunConfig(project="test-project")
     ray_config = FinetuningRayConfig(num_workers=1, use_gpu=False)
     return FinetuningJobConfig(
         name="test-job",

diff --git a/tests/integration/test_lm_harness.py b/tests/integration/test_lm_harness.py
@@ -10,7 +10,7 @@
 @pytest.fixture
 def job_config(llm_model_path) -> LMHarnessJobConfig:
     model_config = AutoModelConfig(path=format_file_path(llm_model_path))
-    tracking_config = WandbRunConfig(name="test-lm-harness-job")
+    tracking_config = WandbRunConfig(project="test-project")
     evaluation_config = LMHarnessEvaluationConfig(tasks=["hellaswag"], limit=5)
     return LMHarnessJobConfig(
         name="test-job",