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
awx_zipline_ai-0.3.1.dist-info/RECORD ADDED
@@ -0,0 +1,96 @@
1
+ __init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
+ agent/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
3
+ agent/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
4
+ agent/ttypes.py,sha256=264pSOgpRPm6XMjdwxdi9W7ciTYElPby7smK7z18RBQ,64304
5
+ ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
+ ai/chronon/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
+ ai/chronon/airflow_helpers.py,sha256=MVkSIrpBMx3XqvAORxcuiwaLdQDijV1yFT5j1PSkNWw,8760
8
+ ai/chronon/constants.py,sha256=74zhArZB-h9jWcNe41kQr2GEC5qgcSw_V9yIwwW0O_E,165
9
+ ai/chronon/group_by.py,sha256=ubrbJrKyYqZsJY2O4UMKXpHanNgcD789lHl-S3XXnjU,27735
10
+ ai/chronon/join.py,sha256=VM8qG1_IDwXprBtLjm3zSZJjQ57MZPlpXue5dJdjOjs,25817
11
+ ai/chronon/logger.py,sha256=rPyeV8uAFvJRY57vJg9KFyEjE4XMl8nVYEoHN1y2dmY,861
12
+ ai/chronon/model.py,sha256=UVF0_-NF2Zdy5zLHXDibFzKMfuMfmYCcOK80cnT7vzA,1115
13
+ ai/chronon/query.py,sha256=AUgUwVeGWH2DKldMa2gt34DiZI3GUM3PtnY26DYgeNs,5417
14
+ ai/chronon/source.py,sha256=xl3Gb925893nSEFuFkWXpuk10kQySyR3XePMNpf8WMs,4016
15
+ ai/chronon/staging_query.py,sha256=Fv4BLEhYjx-YYZbMmBhwlQBTPfEXgUNLBZpTkj9qBvU,8519
16
+ ai/chronon/types.py,sha256=ppH8i3PYVicOQE3h5DWsvtEFTTVnoEAcW8tn7Htc89E,1501
17
+ ai/chronon/utils.py,sha256=D1idc9Y2YYb3Qf2K-CQhh4kGm5Glc0vRSe141y89DPs,15701
18
+ ai/chronon/windows.py,sha256=gvtp1-XHu9yp9xkOy4oDaTGDn7GfIa0JhFXb_RBm12Q,1333
19
+ ai/chronon/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
20
+ ai/chronon/cli/git_utils.py,sha256=6uez-TOvuz9rsUcyq9JYTITvwJy9I-JVZAyX2CNdki8,4255
21
+ ai/chronon/cli/logger.py,sha256=2VVleE_tt4E9ysLHRR3bckfSWEhFugjwOIVB6KYFWDE,1437
22
+ ai/chronon/cli/compile/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
23
+ ai/chronon/cli/compile/column_hashing.py,sha256=9aMCJo0ITt4GUNStajo4evkU2zGmH8XtDOJJuLqGvjg,13760
24
+ ai/chronon/cli/compile/compile_context.py,sha256=DFINBU2l5ISenPlP2rSme1-tRd5svYVM8l0iOEv1nQ0,6648
25
+ ai/chronon/cli/compile/compiler.py,sha256=GUQblOz-fMCZ4b-VAiS1FGocjnlBH7nexw7X_Py2TXY,6913
26
+ ai/chronon/cli/compile/conf_validator.py,sha256=ezzvKeJykvFTzCWoMZF8E4qUUFRkGLCgTBW_ofXGYgY,29058
27
+ ai/chronon/cli/compile/fill_templates.py,sha256=LHZ6xZMyplm1rqr56ziiZpyu5RJOhNtsbLrKC-kyPg4,1204
28
+ ai/chronon/cli/compile/parse_configs.py,sha256=xiaM4fX3Ubxpjya6MphMHd045bMu8fyjZ8zbeTo2DWk,4292
29
+ ai/chronon/cli/compile/parse_teams.py,sha256=x8LyHJGcbv_iDSqyvev-ll4mo96KXpfoOfb8ND7rmDY,7962
30
+ ai/chronon/cli/compile/serializer.py,sha256=Qky6AhrRTxhyNSKXD5_Jp6VecNUdmGy32tCL8BtKBY4,4272
31
+ ai/chronon/cli/compile/version_utils.py,sha256=1T6HamqMkfk-zel4E0-0BIn5I82mumC98FimFu_zjgs,1220
32
+ ai/chronon/cli/compile/display/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
33
+ ai/chronon/cli/compile/display/class_tracker.py,sha256=QEn2pySXDMaNTFfigZW2aPRyYLF38vrru9yZ1QohlW4,3671
34
+ ai/chronon/cli/compile/display/compile_status.py,sha256=rvQ3CecKbIx_3VabFcJnwZyhLK_Bo5GEoSDXGSe1Ca4,3094
35
+ ai/chronon/cli/compile/display/compiled_obj.py,sha256=hsstubchqQDBcFmarRLd4DgMt_sUvTmDlwmNBP01-VI,217
36
+ ai/chronon/cli/compile/display/console.py,sha256=UpMqeJ0C8i0pkue1AHnnyyX0bFJ9zZeJ7HBR6yhuA8A,54
37
+ ai/chronon/cli/compile/display/diff_result.py,sha256=VOkCqImR2TrhUmuWea4GDo3VqkMnVHmbmoLmR-6uS6U,4120
38
+ ai/chronon/repo/__init__.py,sha256=HQ60TFAvCI9uO164uBggIOMs99KPWd_1gae_mMJIfgQ,1413
39
+ ai/chronon/repo/aws.py,sha256=iaFaE-J31Du4hNzx9vknOX5j-A-1OfzeGBhstO2FogE,11745
40
+ ai/chronon/repo/cluster.py,sha256=_dUFMPmjgzRQSSKJVP2qwyGfQLPw0BiUnlUNRPqJY1c,4972
41
+ ai/chronon/repo/compile.py,sha256=bJCDe0Ubv74eOkVDfSeJ3BMcb6yI03Wy4jvzQBivQH4,1820
42
+ ai/chronon/repo/constants.py,sha256=JRDjvV2ZOSIxWJfXmnBngvtRIDBiU8A7869yTzZDiJc,5075
43
+ ai/chronon/repo/default_runner.py,sha256=u2mJrw1xZc8ECK_v9gQSshcVrJDmBb8L9oo7UnuoMAQ,11180
44
+ ai/chronon/repo/explore.py,sha256=rmpUwODc69Gdvx9rYj8vx-Xg5UWD9k1TvBDrs716oUE,14584
45
+ ai/chronon/repo/extract_objects.py,sha256=cyXX07icGTHUzrw__SeQpO76rPMt7jHFnEW08lz56io,4761
46
+ ai/chronon/repo/gcp.py,sha256=z4fTxe0D8z22O52888XvooL-yxMGhNq8aBcTw5Jr1-A,25117
47
+ ai/chronon/repo/gitpython_utils.py,sha256=HIcQNFMf68G0qHvIbk7HSwkcqFA97YrcMsgXw5qw-ow,435
48
+ ai/chronon/repo/hub_runner.py,sha256=jU3x3cwbsClWvmg2eyOMCDLOV6RhbIedKs89IAgP5Wc,8920
49
+ ai/chronon/repo/hub_uploader.py,sha256=2oAAnIru1d7mDq5xAYRt7KWlZ0o_YhQujfGpsL5ph9E,3995
50
+ ai/chronon/repo/init.py,sha256=9Xhf4oeKuDtxpiS6kNf6rOju8-u7VyQDy3BTiMoSaS4,1740
51
+ ai/chronon/repo/join_backfill.py,sha256=WLN9TY6bpupd5T5BFcrTH6qSqpy39P8O9ZvrEhs9vzQ,4024
52
+ ai/chronon/repo/run.py,sha256=Kex79ebLH0bDdq1k1dBK1EC7Y8691trMsvzrTYRQtVs,10780
53
+ ai/chronon/repo/serializer.py,sha256=-nLxTIT-YR7evDCCTAsL_8NusrRz-wFwn-AcBFF7Ocw,4979
54
+ ai/chronon/repo/team_json_utils.py,sha256=JsUc_KwTFn4VMn1ZYwJAPTIeSOfUsDoywzTsBr9Oi70,1472
55
+ ai/chronon/repo/utils.py,sha256=iKJt-qzV3GNLo8yTnc-X7NzKIpweozpS777NddCA74Q,19148
56
+ ai/chronon/repo/zipline.py,sha256=ktWaSbvto3FK5LmB5lsbztOlhcYEbNeVGmznPMk8R3w,1050
57
+ ai/chronon/repo/zipline_hub.py,sha256=laNoRdu_WrAbwE62YMCb7S2pxV4J5JZ0laKW4Za7mTs,12889
58
+ ai/chronon/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
59
+ ai/chronon/resources/gcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
60
+ ai/chronon/resources/gcp/teams.py,sha256=UBSSZjvSYf4MnujqJgyxmDUbjX8KZS8HBoPyn29SH9o,2505
61
+ ai/chronon/resources/gcp/group_bys/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
62
+ ai/chronon/resources/gcp/group_bys/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
63
+ ai/chronon/resources/gcp/group_bys/test/data.py,sha256=UG6_q9aKXiGFX2P4BOSIL6UIQjT-4wHpWXl96UaQcnk,1089
64
+ ai/chronon/resources/gcp/joins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
65
+ ai/chronon/resources/gcp/joins/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
66
+ ai/chronon/resources/gcp/joins/test/data.py,sha256=mhDpA_zKHiHNwpsku8AqsNL-WDA_LH81CO405tIZ9Og,784
67
+ ai/chronon/resources/gcp/sources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
68
+ ai/chronon/resources/gcp/sources/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
69
+ ai/chronon/resources/gcp/sources/test/data.py,sha256=Im-tN2ReJ_K1syRX6IXKoICNwv45tvynBgqTL0qkluQ,1092
70
+ gen_thrift/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
71
+ gen_thrift/api/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
72
+ gen_thrift/api/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
73
+ gen_thrift/api/ttypes.py,sha256=1AGnaSImuXHAVdL6IldrQmBfBW-5Or-D64BgIT7EfHU,147829
74
+ gen_thrift/common/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
75
+ gen_thrift/common/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
76
+ gen_thrift/common/ttypes.py,sha256=zvxBcnY6OPaioA3LepUgQIxaHCsNTuPUuxbtxwjR9u8,58939
77
+ gen_thrift/eval/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
78
+ gen_thrift/eval/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
79
+ gen_thrift/eval/ttypes.py,sha256=uky--UKz-2vuN6Z7QN7ctEqoyPx69GsfOBZAtIdN79M,28560
80
+ gen_thrift/fetcher/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
81
+ gen_thrift/fetcher/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
82
+ gen_thrift/fetcher/ttypes.py,sha256=XdTAL7ql_bQQT4M_eJqs9yOecE6gqa6gO5RoCwsRGWE,4553
83
+ gen_thrift/hub/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
84
+ gen_thrift/hub/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
85
+ gen_thrift/hub/ttypes.py,sha256=tn1VJ3y-sS3NN_gRttTHWmnNhdbSGb4rypApT4NCdBQ,39453
86
+ gen_thrift/observability/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
87
+ gen_thrift/observability/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
88
+ gen_thrift/observability/ttypes.py,sha256=QTKvQ9zqgbj4UcujIs8LuvDXvuyONWTrtsoiQB6ULxE,82254
89
+ gen_thrift/planner/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
90
+ gen_thrift/planner/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
91
+ gen_thrift/planner/ttypes.py,sha256=EXfaw3bPwax8Bl4EfF9b-5KwQW0-9kby4NCiZnIIZZ8,67348
92
+ awx_zipline_ai-0.3.1.dist-info/METADATA,sha256=J2J6VFocuseNFtAjl_H11vexsQQVu32-l6RbjPkzE60,5789
93
+ awx_zipline_ai-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
94
+ awx_zipline_ai-0.3.1.dist-info/entry_points.txt,sha256=EZGwU6w_Ve4BZ2z8zYrIJW0zMMon0IK2FIvIhoSzeqE,60
95
+ awx_zipline_ai-0.3.1.dist-info/top_level.txt,sha256=PY-p-EGZt3eyJqEK5pzRx38U921A1_aRWyCS0a0blvQ,29
96
+ awx_zipline_ai-0.3.1.dist-info/RECORD,,
awx_zipline_ai-0.3.1.dist-info/top_level.txt ADDED
@@ -0,0 +1,4 @@
1
+ __init__
2
+ agent
3
+ ai
4
+ gen_thrift
gen_thrift/__init__.py ADDED
File without changes