prettify test file with test for custom judge model (vertexai)

Author: SecroLoL

e2etests/judgment_client_test.py

@@ -199,7 +199,7 @@ def test_classifier_scorer(client: JudgmentClient):
     print(res)
 
 
-def test_custom_judge(client: JudgmentClient):
+def test_custom_judge_vertexai(client: JudgmentClient):
 
     import vertexai
     from vertexai.generative_models import GenerativeModel
@@ -281,4 +281,9 @@ def get_model_name(self) -> str:
     print("Classifier scorer test successful")
     print("*" * 40)
 
+    print("Testing custom judge")
+    test_custom_judge_vertexai(ui_client)
+    print("Custom judge test successful")
+    print("*" * 40)
+
     print("All tests passed successfully")