add: test for eval classification

.github/workflows/tests.yaml

@@ -66,3 +66,17 @@ jobs:
           pip install -e ".[dev]"
           huggingface-cli login --token ${{ secrets.HF_TOKEN }}
           python -m pytest -s tests/test_prompt_injection_classifier.py
+  
+  test-evaluation-classification:
+    name: test evaluation classification
+    runs-on: ubuntu-latest
+    env:
+      WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
+    steps:
+      - uses: actions/checkout@v4
+      - run: |
+          pip install -U pip
+          pip install torch --index-url https://download.pytorch.org/whl/cpu
+          pip install -e ".[dev]"
+          wandb login --relogin
+          python -m pytest -s tests/test_evaluation_classification.py

tests/test_evaluation_classification.py

@@ -0,0 +1,12 @@
+from safeguards.meta_evaluation import EvaluationClassifier
+
+
+def test_evaluation_classification():
+    parser = EvaluationClassifier(
+        project="geekyrakshit/guardrails-genie",
+        call_id="0193bc3f-cb02-7271-89bd-e7fdccfc4edc",
+    )
+    parser.register_predict_and_score_calls(
+        max_predict_and_score_calls=2, save_filepath="evaluation.json"
+    )
+    assert len(parser.predict_and_score_calls) == 2