add: wandb job_type

guardrails_genie/train/llama_guard.py

@@ -213,6 +213,7 @@ class LlamaGuardFineTuner:
             project=self.wandb_project,
             entity=self.wandb_entity,
             name=f"{self.model_name}-{self.dataset_name}",
+            job_type="fine-tune-llama-guard",
         )
         self.model.classifier = nn.Linear(
             self.model.classifier.in_features, num_classes

guardrails_genie/train/train_classifier.py

@@ -62,7 +62,12 @@ def train_binary_classifier(
         Exception: If an error occurs during training, the exception is raised after
             ensuring Weights & Biases run is finished.
     """
-    wandb.init(project=project_name, entity=entity_name, name=run_name)
+    wandb.init(
+        project=project_name,
+        entity=entity_name,
+        name=run_name,
+        job_type="train-binary-classifier",
+    )
     if streamlit_mode:
         st.markdown(
             f"Explore your training logs on [Weights & Biases]({wandb.run.url})"