awx-zipline-ai 0.2.1__py3-none-any.whl → 0.3.1__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. agent/ttypes.py +6 -6
  2. ai/chronon/airflow_helpers.py +20 -23
  3. ai/chronon/cli/__init__.py +0 -0
  4. ai/chronon/cli/compile/__init__.py +0 -0
  5. ai/chronon/cli/compile/column_hashing.py +40 -17
  6. ai/chronon/cli/compile/compile_context.py +13 -17
  7. ai/chronon/cli/compile/compiler.py +59 -36
  8. ai/chronon/cli/compile/conf_validator.py +251 -99
  9. ai/chronon/cli/compile/display/__init__.py +0 -0
  10. ai/chronon/cli/compile/display/class_tracker.py +6 -16
  11. ai/chronon/cli/compile/display/compile_status.py +10 -10
  12. ai/chronon/cli/compile/display/diff_result.py +79 -14
  13. ai/chronon/cli/compile/fill_templates.py +3 -8
  14. ai/chronon/cli/compile/parse_configs.py +10 -17
  15. ai/chronon/cli/compile/parse_teams.py +38 -34
  16. ai/chronon/cli/compile/serializer.py +3 -9
  17. ai/chronon/cli/compile/version_utils.py +42 -0
  18. ai/chronon/cli/git_utils.py +2 -13
  19. ai/chronon/cli/logger.py +0 -2
  20. ai/chronon/constants.py +1 -1
  21. ai/chronon/group_by.py +47 -47
  22. ai/chronon/join.py +46 -32
  23. ai/chronon/logger.py +1 -2
  24. ai/chronon/model.py +9 -4
  25. ai/chronon/query.py +2 -2
  26. ai/chronon/repo/__init__.py +1 -2
  27. ai/chronon/repo/aws.py +17 -31
  28. ai/chronon/repo/cluster.py +121 -50
  29. ai/chronon/repo/compile.py +14 -8
  30. ai/chronon/repo/constants.py +1 -1
  31. ai/chronon/repo/default_runner.py +32 -54
  32. ai/chronon/repo/explore.py +70 -73
  33. ai/chronon/repo/extract_objects.py +6 -9
  34. ai/chronon/repo/gcp.py +89 -88
  35. ai/chronon/repo/gitpython_utils.py +3 -2
  36. ai/chronon/repo/hub_runner.py +145 -55
  37. ai/chronon/repo/hub_uploader.py +2 -1
  38. ai/chronon/repo/init.py +12 -5
  39. ai/chronon/repo/join_backfill.py +19 -5
  40. ai/chronon/repo/run.py +42 -39
  41. ai/chronon/repo/serializer.py +4 -12
  42. ai/chronon/repo/utils.py +72 -63
  43. ai/chronon/repo/zipline.py +3 -19
  44. ai/chronon/repo/zipline_hub.py +211 -39
  45. ai/chronon/resources/__init__.py +0 -0
  46. ai/chronon/resources/gcp/__init__.py +0 -0
  47. ai/chronon/resources/gcp/group_bys/__init__.py +0 -0
  48. ai/chronon/resources/gcp/group_bys/test/data.py +13 -17
  49. ai/chronon/resources/gcp/joins/__init__.py +0 -0
  50. ai/chronon/resources/gcp/joins/test/data.py +4 -8
  51. ai/chronon/resources/gcp/sources/__init__.py +0 -0
  52. ai/chronon/resources/gcp/sources/test/data.py +9 -6
  53. ai/chronon/resources/gcp/teams.py +9 -21
  54. ai/chronon/source.py +2 -4
  55. ai/chronon/staging_query.py +60 -19
  56. ai/chronon/types.py +3 -2
  57. ai/chronon/utils.py +21 -68
  58. ai/chronon/windows.py +2 -4
  59. {awx_zipline_ai-0.2.1.dist-info → awx_zipline_ai-0.3.1.dist-info}/METADATA +48 -24
  60. awx_zipline_ai-0.3.1.dist-info/RECORD +96 -0
  61. awx_zipline_ai-0.3.1.dist-info/top_level.txt +4 -0
  62. gen_thrift/__init__.py +0 -0
  63. {ai/chronon → gen_thrift}/api/ttypes.py +327 -197
  64. {ai/chronon/api → gen_thrift}/common/ttypes.py +9 -39
  65. gen_thrift/eval/ttypes.py +660 -0
  66. {ai/chronon → gen_thrift}/hub/ttypes.py +12 -131
  67. {ai/chronon → gen_thrift}/observability/ttypes.py +343 -180
  68. {ai/chronon → gen_thrift}/planner/ttypes.py +326 -45
  69. ai/chronon/eval/__init__.py +0 -122
  70. ai/chronon/eval/query_parsing.py +0 -19
  71. ai/chronon/eval/sample_tables.py +0 -100
  72. ai/chronon/eval/table_scan.py +0 -186
  73. ai/chronon/orchestration/ttypes.py +0 -4406
  74. ai/chronon/resources/gcp/README.md +0 -174
  75. ai/chronon/resources/gcp/zipline-cli-install.sh +0 -54
  76. awx_zipline_ai-0.2.1.dist-info/RECORD +0 -93
  77. awx_zipline_ai-0.2.1.dist-info/licenses/LICENSE +0 -202
  78. awx_zipline_ai-0.2.1.dist-info/top_level.txt +0 -3
  79. /jars/__init__.py → /__init__.py +0 -0
  80. {awx_zipline_ai-0.2.1.dist-info → awx_zipline_ai-0.3.1.dist-info}/WHEEL +0 -0
  81. {awx_zipline_ai-0.2.1.dist-info → awx_zipline_ai-0.3.1.dist-info}/entry_points.txt +0 -0
  82. {ai/chronon → gen_thrift}/api/__init__.py +0 -0
  83. {ai/chronon/api/common → gen_thrift/api}/constants.py +0 -0
  84. {ai/chronon/api → gen_thrift}/common/__init__.py +0 -0
  85. {ai/chronon/api → gen_thrift/common}/constants.py +0 -0
  86. {ai/chronon/fetcher → gen_thrift/eval}/__init__.py +0 -0
  87. {ai/chronon/fetcher → gen_thrift/eval}/constants.py +0 -0
  88. {ai/chronon/hub → gen_thrift/fetcher}/__init__.py +0 -0
  89. {ai/chronon/hub → gen_thrift/fetcher}/constants.py +0 -0
  90. {ai/chronon → gen_thrift}/fetcher/ttypes.py +0 -0
  91. {ai/chronon/observability → gen_thrift/hub}/__init__.py +0 -0
  92. {ai/chronon/observability → gen_thrift/hub}/constants.py +0 -0
  93. {ai/chronon/orchestration → gen_thrift/observability}/__init__.py +0 -0
  94. {ai/chronon/orchestration → gen_thrift/observability}/constants.py +0 -0
  95. {ai/chronon → gen_thrift}/planner/__init__.py +0 -0
  96. {ai/chronon → gen_thrift}/planner/constants.py +0 -0
{ai/chronon/api → gen_thrift}/common/ttypes.py RENAMED
@@ -211,16 +211,8 @@ class EnvironmentVariables(object):
211
211
  thrift_spec = None
212
212
 
213
213
 
214
- def __init__(self, common = {
215
- }, modeEnvironments = {
216
- },):
217
- if common is self.thrift_spec[1][4]:
218
- common = {
219
- }
214
+ def __init__(self, common = None, modeEnvironments = None,):
220
215
  self.common = common
221
- if modeEnvironments is self.thrift_spec[2][4]:
222
- modeEnvironments = {
223
- }
224
216
  self.modeEnvironments = modeEnvironments
225
217
 
226
218
  def read(self, iprot):
@@ -330,16 +322,8 @@ class ConfigProperties(object):
330
322
  thrift_spec = None
331
323
 
332
324
 
333
- def __init__(self, common = {
334
- }, modeConfigs = {
335
- },):
336
- if common is self.thrift_spec[1][4]:
337
- common = {
338
- }
325
+ def __init__(self, common = None, modeConfigs = None,):
339
326
  self.common = common
340
- if modeConfigs is self.thrift_spec[2][4]:
341
- modeConfigs = {
342
- }
343
327
  self.modeConfigs = modeConfigs
344
328
 
345
329
  def read(self, iprot):
@@ -449,16 +433,8 @@ class ClusterConfigProperties(object):
449
433
  thrift_spec = None
450
434
 
451
435
 
452
- def __init__(self, common = {
453
- }, modeClusterConfigs = {
454
- },):
455
- if common is self.thrift_spec[1][4]:
456
- common = {
457
- }
436
+ def __init__(self, common = None, modeClusterConfigs = None,):
458
437
  self.common = common
459
- if modeClusterConfigs is self.thrift_spec[2][4]:
460
- modeClusterConfigs = {
461
- }
462
438
  self.modeClusterConfigs = modeClusterConfigs
463
439
 
464
440
  def read(self, iprot):
@@ -1173,26 +1149,20 @@ DateRange.thrift_spec = (
1173
1149
  all_structs.append(EnvironmentVariables)
1174
1150
  EnvironmentVariables.thrift_spec = (
1175
1151
  None, # 0
1176
- (1, TType.MAP, 'common', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), {
1177
- }, ), # 1
1178
- (2, TType.MAP, 'modeEnvironments', (TType.STRING, 'UTF8', TType.MAP, (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), False), {
1179
- }, ), # 2
1152
+ (1, TType.MAP, 'common', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), None, ), # 1
1153
+ (2, TType.MAP, 'modeEnvironments', (TType.STRING, 'UTF8', TType.MAP, (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), False), None, ), # 2
1180
1154
  )
1181
1155
  all_structs.append(ConfigProperties)
1182
1156
  ConfigProperties.thrift_spec = (
1183
1157
  None, # 0
1184
- (1, TType.MAP, 'common', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), {
1185
- }, ), # 1
1186
- (2, TType.MAP, 'modeConfigs', (TType.STRING, 'UTF8', TType.MAP, (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), False), {
1187
- }, ), # 2
1158
+ (1, TType.MAP, 'common', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), None, ), # 1
1159
+ (2, TType.MAP, 'modeConfigs', (TType.STRING, 'UTF8', TType.MAP, (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), False), None, ), # 2
1188
1160
  )
1189
1161
  all_structs.append(ClusterConfigProperties)
1190
1162
  ClusterConfigProperties.thrift_spec = (
1191
1163
  None, # 0
1192
- (1, TType.MAP, 'common', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), {
1193
- }, ), # 1
1194
- (2, TType.MAP, 'modeClusterConfigs', (TType.STRING, 'UTF8', TType.MAP, (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), False), {
1195
- }, ), # 2
1164
+ (1, TType.MAP, 'common', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), None, ), # 1
1165
+ (2, TType.MAP, 'modeClusterConfigs', (TType.STRING, 'UTF8', TType.MAP, (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), False), None, ), # 2
1196
1166
  )
1197
1167
  all_structs.append(TableInfo)
1198
1168
  TableInfo.thrift_spec = (