work with .keys file

neurostuff · Nov 22, 2024 · 8c5237f · 8c5237f
1 parent 35c09aa
commit 8c5237f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -31,4 +31,6 @@ jobs:
       - name: Test with pytest
         env:
           OPENAI_API_KEY: "fake_key"
-        run: pytest
+        run: |
+          cp .keys.example .keys
+          pytest
diff --git a/.gitignore b/.gitignore
@@ -72,6 +72,7 @@ ipython_config.py
 dmypy.json
 
 # Environments
+.keys
 .env
 .venv
 env/

diff --git a/.keys.example b/.keys.example
@@ -0,0 +1 @@
+OPENAI_CLIENT_API_KEY=fake_key
diff --git a/tests/test_participant_demographics.py b/tests/test_participant_demographics.py
@@ -9,10 +9,10 @@
 def test_ParticipantDemographicsExtractor(sample_data, tmp_path):
     """Test the word count extraction pipeline."""
     pde = ParticipantDemographicsExtractor(
-        extraction_model="gpt-4o-2024-08-06",
+        extraction_model="gpt-4o-mini-2024-07-18",
         prompt_set="ZERO_SHOT_MULTI_GROUP_FTSTRICT_FC",
         env_variable="API_CLIENT_OPENAI_KEY",
-        env_file=str(Path(__file__).parents[1] / ".env"),
+        env_file=str(Path(__file__).parents[1] / ".keys"),
     )
     dataset = Dataset(sample_data)
     output_dir = tmp_path / "participant_demographics"
-Original file line number
+Diff line change
@@ Expand Up / @@ -72,6 +72,7 @@ ipython_config.py @@
     dmypy.json
     # Environments
+    .keys
     .env
     .venv
     env/
@@ Expand Down @@