make test-creation evaluation to dump json instead of pickle

UBC-MDS · Jun 6, 2024 · c124566 · c124566
1 parent 4eb9c2d
commit c124566
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/test_creation/cli/repository.py b/src/test_creation/cli/repository.py
@@ -71,8 +71,11 @@ def evaluate(repo_path: str, report_output_path: str = None,
                                          repository=repo, checklist=checklist)
         response = evaluator.run()
         if response_output_path:
-            with open(response_output_path, 'wb') as file:
-                pickle.dump(response, file)
+            # with open(response_output_path+".pickle", 'wb') as file:
+            #     pickle.dump(response, file)
+            json_str = response.model_dump_json()
+            with open(response_output_path, 'w') as f:
+                f.write(json_str)
 
         if report_output_path:
             parser = ResponseParser(response)