From 03084844dd6c59cb1ec369026d107a167ccb3d19 Mon Sep 17 00:00:00 2001
From: Daniel Bachhuber <daniel.b@posthog.com>
Date: Thu, 19 Dec 2024 15:16:57 -0800
Subject: [PATCH] Resolve various warnings

---
 .../experiments/test/test_experiment_trends_query_runner.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/posthog/hogql_queries/experiments/test/test_experiment_trends_query_runner.py b/posthog/hogql_queries/experiments/test/test_experiment_trends_query_runner.py
index b52bb3bf39727..d7503974c0cdd 100644
--- a/posthog/hogql_queries/experiments/test/test_experiment_trends_query_runner.py
+++ b/posthog/hogql_queries/experiments/test/test_experiment_trends_query_runner.py
@@ -91,8 +91,12 @@ def create_experiment(
             feature_flag = self.create_feature_flag(name)
         if start_date is None:
             start_date = timezone.now()
+        else:
+            start_date = timezone.make_aware(start_date)  # Make naive datetime timezone-aware
         if end_date is None:
             end_date = timezone.now() + timedelta(days=14)
+        elif end_date is not None:
+            end_date = timezone.make_aware(end_date)  # Make naive datetime timezone-aware
         return Experiment.objects.create(
             name=name,
             team=self.team,
@@ -147,7 +151,6 @@ def create_data_warehouse_table_with_payments(self):
             filesystem=fs,
             use_dictionary=True,
             compression="snappy",
-            version="2.0",
         )
 
         table_name = "payments"
@@ -212,7 +215,6 @@ def create_data_warehouse_table_with_usage(self):
             filesystem=fs,
             use_dictionary=True,
             compression="snappy",
-            version="2.0",
         )
 
         table_name = "usage"