Merge pull request #21 from duml/main

chore: fixed "You must set output_field" bug
jamessewell · Sep 1, 2021 · f9f12d0 · f9f12d0
2 parents 5c15242 + 5d98d6e
commit f9f12d0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/timescale/db/models/expressions.py b/timescale/db/models/expressions.py
@@ -6,7 +6,7 @@
 )
 from django.utils import timezone
 from datetime import timedelta
-
+from timescale.db.models.fields import TimescaleDateTimeField
 
 class Interval(models.Func):
     """
@@ -46,7 +46,8 @@ class TimeBucket(models.Func):
     def __init__(self, expression, interval, *args, **kwargs):
         if not isinstance(interval, models.Value):
             interval = models.Value(interval)
-        super().__init__(interval, expression, *args, **kwargs)
+        output_field = TimescaleDateTimeField(interval=interval)
+        super().__init__(interval, expression, output_field=output_field)
 
 
 class TimeBucketGapFill(models.Func):
@@ -73,4 +74,5 @@ def __init__(
     ):
         if not isinstance(interval, models.Value):
             interval = Interval(interval) / datapoints
-        super().__init__(interval, expression, start, end, *args, **kwargs)
+        output_field = TimescaleDateTimeField(interval=interval)
+        super().__init__(interval, expression, start, end, output_field=output_field)