awx-zipline-ai 0.0.32__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (96) hide show
  1. __init__.py +0 -0
  2. agent/__init__.py +1 -0
  3. agent/constants.py +15 -0
  4. agent/ttypes.py +1684 -0
  5. ai/__init__.py +0 -0
  6. ai/chronon/__init__.py +0 -0
  7. ai/chronon/airflow_helpers.py +248 -0
  8. ai/chronon/cli/__init__.py +0 -0
  9. ai/chronon/cli/compile/__init__.py +0 -0
  10. ai/chronon/cli/compile/column_hashing.py +336 -0
  11. ai/chronon/cli/compile/compile_context.py +173 -0
  12. ai/chronon/cli/compile/compiler.py +183 -0
  13. ai/chronon/cli/compile/conf_validator.py +742 -0
  14. ai/chronon/cli/compile/display/__init__.py +0 -0
  15. ai/chronon/cli/compile/display/class_tracker.py +102 -0
  16. ai/chronon/cli/compile/display/compile_status.py +95 -0
  17. ai/chronon/cli/compile/display/compiled_obj.py +12 -0
  18. ai/chronon/cli/compile/display/console.py +3 -0
  19. ai/chronon/cli/compile/display/diff_result.py +111 -0
  20. ai/chronon/cli/compile/fill_templates.py +35 -0
  21. ai/chronon/cli/compile/parse_configs.py +134 -0
  22. ai/chronon/cli/compile/parse_teams.py +242 -0
  23. ai/chronon/cli/compile/serializer.py +109 -0
  24. ai/chronon/cli/compile/version_utils.py +42 -0
  25. ai/chronon/cli/git_utils.py +145 -0
  26. ai/chronon/cli/logger.py +59 -0
  27. ai/chronon/constants.py +3 -0
  28. ai/chronon/group_by.py +692 -0
  29. ai/chronon/join.py +580 -0
  30. ai/chronon/logger.py +23 -0
  31. ai/chronon/model.py +40 -0
  32. ai/chronon/query.py +126 -0
  33. ai/chronon/repo/__init__.py +39 -0
  34. ai/chronon/repo/aws.py +284 -0
  35. ai/chronon/repo/cluster.py +136 -0
  36. ai/chronon/repo/compile.py +62 -0
  37. ai/chronon/repo/constants.py +164 -0
  38. ai/chronon/repo/default_runner.py +269 -0
  39. ai/chronon/repo/explore.py +418 -0
  40. ai/chronon/repo/extract_objects.py +134 -0
  41. ai/chronon/repo/gcp.py +586 -0
  42. ai/chronon/repo/gitpython_utils.py +15 -0
  43. ai/chronon/repo/hub_runner.py +261 -0
  44. ai/chronon/repo/hub_uploader.py +109 -0
  45. ai/chronon/repo/init.py +60 -0
  46. ai/chronon/repo/join_backfill.py +119 -0
  47. ai/chronon/repo/run.py +296 -0
  48. ai/chronon/repo/serializer.py +133 -0
  49. ai/chronon/repo/team_json_utils.py +46 -0
  50. ai/chronon/repo/utils.py +481 -0
  51. ai/chronon/repo/zipline.py +35 -0
  52. ai/chronon/repo/zipline_hub.py +277 -0
  53. ai/chronon/resources/__init__.py +0 -0
  54. ai/chronon/resources/gcp/__init__.py +0 -0
  55. ai/chronon/resources/gcp/group_bys/__init__.py +0 -0
  56. ai/chronon/resources/gcp/group_bys/test/__init__.py +0 -0
  57. ai/chronon/resources/gcp/group_bys/test/data.py +30 -0
  58. ai/chronon/resources/gcp/joins/__init__.py +0 -0
  59. ai/chronon/resources/gcp/joins/test/__init__.py +0 -0
  60. ai/chronon/resources/gcp/joins/test/data.py +26 -0
  61. ai/chronon/resources/gcp/sources/__init__.py +0 -0
  62. ai/chronon/resources/gcp/sources/test/__init__.py +0 -0
  63. ai/chronon/resources/gcp/sources/test/data.py +26 -0
  64. ai/chronon/resources/gcp/teams.py +58 -0
  65. ai/chronon/source.py +86 -0
  66. ai/chronon/staging_query.py +226 -0
  67. ai/chronon/types.py +58 -0
  68. ai/chronon/utils.py +510 -0
  69. ai/chronon/windows.py +48 -0
  70. awx_zipline_ai-0.0.32.dist-info/METADATA +197 -0
  71. awx_zipline_ai-0.0.32.dist-info/RECORD +96 -0
  72. awx_zipline_ai-0.0.32.dist-info/WHEEL +5 -0
  73. awx_zipline_ai-0.0.32.dist-info/entry_points.txt +2 -0
  74. awx_zipline_ai-0.0.32.dist-info/top_level.txt +4 -0
  75. gen_thrift/__init__.py +0 -0
  76. gen_thrift/api/__init__.py +1 -0
  77. gen_thrift/api/constants.py +15 -0
  78. gen_thrift/api/ttypes.py +3754 -0
  79. gen_thrift/common/__init__.py +1 -0
  80. gen_thrift/common/constants.py +15 -0
  81. gen_thrift/common/ttypes.py +1814 -0
  82. gen_thrift/eval/__init__.py +1 -0
  83. gen_thrift/eval/constants.py +15 -0
  84. gen_thrift/eval/ttypes.py +660 -0
  85. gen_thrift/fetcher/__init__.py +1 -0
  86. gen_thrift/fetcher/constants.py +15 -0
  87. gen_thrift/fetcher/ttypes.py +127 -0
  88. gen_thrift/hub/__init__.py +1 -0
  89. gen_thrift/hub/constants.py +15 -0
  90. gen_thrift/hub/ttypes.py +1109 -0
  91. gen_thrift/observability/__init__.py +1 -0
  92. gen_thrift/observability/constants.py +15 -0
  93. gen_thrift/observability/ttypes.py +2355 -0
  94. gen_thrift/planner/__init__.py +1 -0
  95. gen_thrift/planner/constants.py +15 -0
  96. gen_thrift/planner/ttypes.py +1967 -0
@@ -0,0 +1,15 @@
1
+ #
2
+ # Autogenerated by Thrift Compiler (0.22.0)
3
+ #
4
+ # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
5
+ #
6
+ # options string: py
7
+ #
8
+
9
+ from thrift.Thrift import TType, TMessageType, TFrozenDict, TException, TApplicationException
10
+ from thrift.protocol.TProtocol import TProtocolException
11
+ from thrift.TRecursive import fix_spec
12
+ from uuid import UUID
13
+
14
+ import sys
15
+ from .ttypes import *
@@ -0,0 +1,127 @@
1
+ #
2
+ # Autogenerated by Thrift Compiler (0.22.0)
3
+ #
4
+ # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
5
+ #
6
+ # options string: py
7
+ #
8
+
9
+ from thrift.Thrift import TType, TMessageType, TFrozenDict, TException, TApplicationException
10
+ from thrift.protocol.TProtocol import TProtocolException
11
+ from thrift.TRecursive import fix_spec
12
+ from uuid import UUID
13
+
14
+ import sys
15
+
16
+ from thrift.transport import TTransport
17
+ all_structs = []
18
+
19
+
20
+ class TileKey(object):
21
+ """
22
+ Attributes:
23
+ - dataset
24
+ - keyBytes
25
+ - tileSizeMillis
26
+ - tileStartTimestampMillis
27
+
28
+ """
29
+ thrift_spec = None
30
+
31
+
32
+ def __init__(self, dataset = None, keyBytes = None, tileSizeMillis = None, tileStartTimestampMillis = None,):
33
+ self.dataset = dataset
34
+ self.keyBytes = keyBytes
35
+ self.tileSizeMillis = tileSizeMillis
36
+ self.tileStartTimestampMillis = tileStartTimestampMillis
37
+
38
+ def read(self, iprot):
39
+ if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
40
+ iprot._fast_decode(self, iprot, [self.__class__, self.thrift_spec])
41
+ return
42
+ iprot.readStructBegin()
43
+ while True:
44
+ (fname, ftype, fid) = iprot.readFieldBegin()
45
+ if ftype == TType.STOP:
46
+ break
47
+ if fid == 1:
48
+ if ftype == TType.STRING:
49
+ self.dataset = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
50
+ else:
51
+ iprot.skip(ftype)
52
+ elif fid == 2:
53
+ if ftype == TType.LIST:
54
+ self.keyBytes = []
55
+ (_etype3, _size0) = iprot.readListBegin()
56
+ for _i4 in range(_size0):
57
+ _elem5 = iprot.readByte()
58
+ self.keyBytes.append(_elem5)
59
+ iprot.readListEnd()
60
+ else:
61
+ iprot.skip(ftype)
62
+ elif fid == 3:
63
+ if ftype == TType.I64:
64
+ self.tileSizeMillis = iprot.readI64()
65
+ else:
66
+ iprot.skip(ftype)
67
+ elif fid == 4:
68
+ if ftype == TType.I64:
69
+ self.tileStartTimestampMillis = iprot.readI64()
70
+ else:
71
+ iprot.skip(ftype)
72
+ else:
73
+ iprot.skip(ftype)
74
+ iprot.readFieldEnd()
75
+ iprot.readStructEnd()
76
+
77
+ def write(self, oprot):
78
+ self.validate()
79
+ if oprot._fast_encode is not None and self.thrift_spec is not None:
80
+ oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
81
+ return
82
+ oprot.writeStructBegin('TileKey')
83
+ if self.dataset is not None:
84
+ oprot.writeFieldBegin('dataset', TType.STRING, 1)
85
+ oprot.writeString(self.dataset.encode('utf-8') if sys.version_info[0] == 2 else self.dataset)
86
+ oprot.writeFieldEnd()
87
+ if self.keyBytes is not None:
88
+ oprot.writeFieldBegin('keyBytes', TType.LIST, 2)
89
+ oprot.writeListBegin(TType.BYTE, len(self.keyBytes))
90
+ for iter6 in self.keyBytes:
91
+ oprot.writeByte(iter6)
92
+ oprot.writeListEnd()
93
+ oprot.writeFieldEnd()
94
+ if self.tileSizeMillis is not None:
95
+ oprot.writeFieldBegin('tileSizeMillis', TType.I64, 3)
96
+ oprot.writeI64(self.tileSizeMillis)
97
+ oprot.writeFieldEnd()
98
+ if self.tileStartTimestampMillis is not None:
99
+ oprot.writeFieldBegin('tileStartTimestampMillis', TType.I64, 4)
100
+ oprot.writeI64(self.tileStartTimestampMillis)
101
+ oprot.writeFieldEnd()
102
+ oprot.writeFieldStop()
103
+ oprot.writeStructEnd()
104
+
105
+ def validate(self):
106
+ return
107
+
108
+ def __repr__(self):
109
+ L = ['%s=%r' % (key, value)
110
+ for key, value in self.__dict__.items()]
111
+ return '%s(%s)' % (self.__class__.__name__, ', '.join(L))
112
+
113
+ def __eq__(self, other):
114
+ return isinstance(other, self.__class__) and self.__dict__ == other.__dict__
115
+
116
+ def __ne__(self, other):
117
+ return not (self == other)
118
+ all_structs.append(TileKey)
119
+ TileKey.thrift_spec = (
120
+ None, # 0
121
+ (1, TType.STRING, 'dataset', 'UTF8', None, ), # 1
122
+ (2, TType.LIST, 'keyBytes', (TType.BYTE, None, False), None, ), # 2
123
+ (3, TType.I64, 'tileSizeMillis', None, None, ), # 3
124
+ (4, TType.I64, 'tileStartTimestampMillis', None, None, ), # 4
125
+ )
126
+ fix_spec(all_structs)
127
+ del all_structs
@@ -0,0 +1 @@
1
+ __all__ = ['ttypes', 'constants']
@@ -0,0 +1,15 @@
1
+ #
2
+ # Autogenerated by Thrift Compiler (0.22.0)
3
+ #
4
+ # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
5
+ #
6
+ # options string: py
7
+ #
8
+
9
+ from thrift.Thrift import TType, TMessageType, TFrozenDict, TException, TApplicationException
10
+ from thrift.protocol.TProtocol import TProtocolException
11
+ from thrift.TRecursive import fix_spec
12
+ from uuid import UUID
13
+
14
+ import sys
15
+ from .ttypes import *