From 7cd768e0e778da88c7f11a806f2b3a19162f39c1 Mon Sep 17 00:00:00 2001 From: aseembits93 Date: Fri, 9 May 2025 19:42:20 -0700 Subject: [PATCH] ready to merge --- django/aiservice/testgen/testgen.py | 2 -- experiments/testgen_analysis.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/django/aiservice/testgen/testgen.py b/django/aiservice/testgen/testgen.py index c7a66c4f9..9427ae48d 100644 --- a/django/aiservice/testgen/testgen.py +++ b/django/aiservice/testgen/testgen.py @@ -365,8 +365,6 @@ async def testgen( return 500, TestGenErrorResponseSchema(error="Error generating tests. Internal server error.") if hasattr(request, "should_log_features") and request.should_log_features: # TODO: Update log features with perf instrumented tests - generated_test_source = "# TRY41 GENERATED WITH SINGLE PROMPT\n" + generated_test_source - instrumented_behavior_test_source = "# TRY41 GENERATED WITH SINGLE PROMPT\n" + instrumented_behavior_test_source await log_features( trace_id=data.trace_id, user_id=request.user, diff --git a/experiments/testgen_analysis.py b/experiments/testgen_analysis.py index a267efe00..de19b6c66 100644 --- a/experiments/testgen_analysis.py +++ b/experiments/testgen_analysis.py @@ -15,7 +15,7 @@ def load_data( with engine.connect() as connection: query = """ SELECT * FROM optimization_features WHERE - user_id = 'github|4725571' AND EXISTS (SELECT 1 FROM unnest(generated_test) AS elem WHERE elem LIKE '%%# TRY21 GENERATED WITH SINGLE PROMPT%%') + user_id = 'github|4725571' AND EXISTS (SELECT 1 FROM unnest(generated_test) AS elem WHERE elem LIKE '%%# TRY41 GENERATED WITH SINGLE PROMPT%%') """ return pd.read_sql_query( query, connection