Merge branch 'master' of github.com:Value4AI/gpv

Value4AI · Sep 25, 2024 · a87f28b · a87f28b
2 parents 7a9de7d + 77ea8dd
commit a87f28b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/gpv/measure.py b/gpv/measure.py
@@ -124,7 +124,7 @@ def measure_texts(self, texts: list[str], values: list[str]):
 
             for value in values:
                 if agg[value]:
-                    agg[value] = np.mean(agg[value])
+                    agg[value] = np.mean(agg[value]).item()
                 else:
                     agg[value] = None
             results["aggregated"] = agg
@@ -286,8 +286,8 @@ def measure_entities_rag(self, text: str, values: list[str], measurement_subject
             subject_value2scores[measurement_subject] = value2scores
 
         # Save value2scores
-        save_path = "value2scores_" + datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + ".json"
-        with open(save_path, "w") as file:
-            json.dump(subject_value2scores, file, indent=4)
+        # save_path = "value2scores_" + datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + ".json"
+        # with open(save_path, "w") as file:
+        #     json.dump(subject_value2scores, file, indent=4)
 
         return subject_value2avg_scores