6
6
)
7
7
from django .utils import timezone
8
8
from datetime import timedelta
9
-
9
+ from timescale . db . models . fields import TimescaleDateTimeField
10
10
11
11
class Interval (models .Func ):
12
12
"""
@@ -46,7 +46,8 @@ class TimeBucket(models.Func):
46
46
def __init__ (self , expression , interval , * args , ** kwargs ):
47
47
if not isinstance (interval , models .Value ):
48
48
interval = models .Value (interval )
49
- super ().__init__ (interval , expression , * args , ** kwargs )
49
+ output_field = TimescaleDateTimeField (interval = interval )
50
+ super ().__init__ (interval , expression , output_field = output_field )
50
51
51
52
52
53
class TimeBucketGapFill (models .Func ):
@@ -73,4 +74,5 @@ def __init__(
73
74
):
74
75
if not isinstance (interval , models .Value ):
75
76
interval = Interval (interval ) / datapoints
76
- super ().__init__ (interval , expression , start , end , * args , ** kwargs )
77
+ output_field = TimescaleDateTimeField (interval = interval )
78
+ super ().__init__ (interval , expression , start , end , output_field = output_field )
0 commit comments