Merge pull request #57 from largelanguagemodels/compare-docstring

Update chatbot_comparator.py

pykoi/component/chatbot_comparator.py

@@ -47,7 +47,7 @@ def __init__(self, models: List[AbsLlm], **kwargs):
                 "latency",
                 "length_in_tokens",
             ]
-        )  # Added this line
+        )
 
     def add(self, model: AbsLlm):
         """
@@ -118,5 +118,13 @@ def inference(self, questions: List[str]):
         return self.inference_results
 
     def visualize(self):
+        """
+        Visualize the inference results using a bar chart.
+
+        Returns
+        -------
+        object
+            A Barchart visualization with the inference results.
+        """
         records = df_to_js_array(self.inference_results)
         return Barchart()(data=records)