update: classifier training

.gitignore

@@ -165,4 +165,6 @@ cursor_prompts/
 uv.lock
 test.py
 temp.txt
-**.csv
+**.csv
+binary-classifier/
+wandb/

guardrails_genie/train_classifier.py

@@ -61,6 +61,8 @@ def train_binary_classifier(
             load_best_model_at_end=True,
             push_to_hub=True,
             report_to="wandb",
+            logging_strategy="steps",
+            logging_steps=1,
         ),
         train_dataset=tokenized_datasets["train"],
         eval_dataset=tokenized_datasets["test"],

pyproject.toml

@@ -12,7 +12,7 @@ dependencies = [
     "ruff>=0.6.9",
     "pip>=24.2",
     "uv>=0.4.20",
-    "git+https://github.com/wandb/weave@feat/eval-progressbar",
+    "weave @ git+https://github.com/wandb/weave@feat/eval-progressbar",
     "streamlit>=1.40.1",
     "python-dotenv>=1.0.1",
     "watchdog>=6.0.0",
@@ -23,4 +23,4 @@ dependencies = [
 ]
 
 [tool.setuptools]
-py-modules = ["guardrails_genie"]
+py-modules = ["guardrails_genie"]