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

awx_zipline_ai-0.0.32.dist-info/METADATA

@@ -0,0 +1,197 @@
+Metadata-Version: 2.4
+Name: awx-zipline-ai
+Version: 0.0.32
+Summary: CLI tool for the Zipline AI platform
+Author-email: Zipline AI <hello@zipline.ai>
+License: Apache License 2.0
+Project-URL: homepage, https://zipline.ai
+Project-URL: documentation, https://docs.zipline.ai
+Project-URL: github, https://github.com/zipline-ai/chronon/
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: boto3==1.40.26
+Requires-Dist: botocore==1.40.26
+Requires-Dist: cachetools==5.5.2
+Requires-Dist: certifi==2025.8.3
+Requires-Dist: charset-normalizer==3.4.3
+Requires-Dist: click==8.2.1
+Requires-Dist: crcmod==1.7
+Requires-Dist: gitdb==4.0.12
+Requires-Dist: gitpython==3.1.45
+Requires-Dist: google-api-core[grpc]==2.25.1
+Requires-Dist: google-auth==2.40.3
+Requires-Dist: google-cloud-bigquery-storage==2.33.0
+Requires-Dist: google-cloud-core==2.4.3
+Requires-Dist: google-cloud-iam==2.19.1
+Requires-Dist: google-cloud-storage==2.19.0
+Requires-Dist: google-crc32c==1.7.1
+Requires-Dist: google-resumable-media==2.7.2
+Requires-Dist: googleapis-common-protos[grpc]==1.70.0
+Requires-Dist: grpc-google-iam-v1==0.14.2
+Requires-Dist: grpcio<=1.74.0,>=1.66.2
+Requires-Dist: grpcio-status<=1.74.0,>=1.62.3
+Requires-Dist: idna==3.10
+Requires-Dist: importlib-resources==6.5.2
+Requires-Dist: jmespath==1.0.1
+Requires-Dist: markdown-it-py==4.0.0
+Requires-Dist: mdurl==0.1.2
+Requires-Dist: proto-plus==1.26.1
+Requires-Dist: protobuf<=6.32.0,>=4.25.5
+Requires-Dist: py4j==0.10.9.7
+Requires-Dist: pyasn1==0.6.1
+Requires-Dist: pyasn1-modules==0.4.2
+Requires-Dist: pygments==2.19.2
+Requires-Dist: pyspark==3.5.4
+Requires-Dist: python-dateutil==2.9.0.post0
+Requires-Dist: requests==2.32.5
+Requires-Dist: rich==14.1.0
+Requires-Dist: rsa==4.9.1
+Requires-Dist: s3transfer==0.13.1
+Requires-Dist: six==1.17.0
+Requires-Dist: smmap==5.0.2
+Requires-Dist: thrift==0.20.0
+Requires-Dist: urllib3==2.5.0
+Requires-Dist: python-dotenv>=1.0.1
+
+### Chronon Python API
+
+
+#### Overview
+
+Chronon Python API for materializing configs to be run by the Chronon Engine. Contains python helpers to help managed a repo of feature and join definitions to be executed by the chronon scala engine.
+
+
+#### User API Overview
+
+##### Sources
+
+Most fields are self explanatory. Time columns are expected to be in milliseconds (unixtime).
+
+```python
+# File <repo>/sources/test_sources.py
+from ai.chronon.query import (
+  Query,
+  select,
+)
+from ai.chronon.api.ttypes import Source, EventSource, EntitySource
+
+# Sample query
+Query(
+  selects=select(
+      user="user_id",
+      created_at="created_at",
+  ),
+  wheres=["has_availability = 1"],
+  start_partition="2021-01-01",  # Defines the beginning of time for computations related to the source.
+  setups=["...UDF..."],
+  time_column="ts",
+  end_partition=None,
+  mutation_time_column="mutation_timestamp",
+  reversal_column="CASE WHEN mutation_type IN ('DELETE', 'UPDATE_BEFORE') THEN true ELSE false END"
+)
+
+user_activity = Source(entities=EntitySource(
+  snapshotTable="db_exports.table",
+  mutationTable="mutations_namespace.table_mutations",
+  mutationTopic="mutationsKafkaTopic",
+  query=Query(...)
+)
+
+website__views = Source(events=EventSource(
+  table="namespace.table",
+  topic="kafkaTopicForEvents",
+)
+```
+
+
+##### Group By (Features)
+
+Group Bys are aggregations over sources that define features. For example:
+
+```python
+# File <repo>/group_bys/example_team/example_group_by.py
+from ai.chronon.group_by import (
+  GroupBy,
+  Window,
+  TimeUnit,
+  Accuracy,
+  Operation,
+  Aggregations,
+  Aggregation,
+  DefaultAggregation,
+)
+from sources import test_sources
+
+sum_cols = [f"active_{x}_days" for x in [30, 90, 120]]
+
+
+v0 = GroupBy(
+  sources=test_source.user_activity,
+  keys=["user"],
+  aggregations=Aggregations(
+    user_active_1_day=Aggregation(operation=Operation.LAST),
+    second_feature=Aggregation(
+      input_column="active_7_days",
+      operation=Operation.SUM,
+      windows=[
+        Window(n, TimeUnit.DAYS) for n in [3, 5, 9]
+      ]
+    ),
+  ) + [
+    Aggregation(
+      input_column=col,
+      operation=Operation.SUM
+    ) for col in sum_columns           # Alternative syntax for defining aggregations.
+  ] + [
+    Aggregation(
+      input_column="device",
+      operation=LAST_K(10)
+    )
+  ],
+  dependencies=[
+    "db_exports.table/ds={{ ds }}"      # If not defined will be derived from the Source info.
+  ],
+  accuracy=Accuracy.SNAPSHOT,          # This could be TEMPORAL for point in time correctness.
+  env={
+    "backfill": {                      # Execution environment variables for each of the modes for `run.py`
+      "EXECUTOR_MEMORY": "4G"
+     },
+  },
+  online=True,                         # True if this group by needs to be uploaded to a KV Store.
+  production=False                     # True if this group by is production level.
+)
+```
+
+##### Join
+
+A Join is a collection of feature values for the keys and (times if applicable) defined on the left (source). Example:
+
+```python
+# File <repo>/joins/example_team/example_join.py
+from ai.chronon.join import Join, JoinPart
+from sources import test_sources
+from group_bys.example_team import example_group_by
+
+v1 = Join(
+    left=test_sources.website__views,
+    right_parts=[
+        JoinPart(group_by=example_group_by.v0),
+    ],
+    online=True,       # True if this join will be fetched in production.
+    production=False,  # True if this join should not use non-production group bys.
+    env={"backfill": {"PARALLELISM": "10"}, "streaming": {"STREAMING_ENV_VAR": "VALUE"}},
+)
+```
+
+##### Pre-commit Setup
+
+1. Install pre-commit and other dev libraries:
+```
+pip install -r requirements/dev.txt
+```
+2. Run the following command under `api/python` to install the git hook scripts:
+```
+pre-commit install
+```
+
+To support more pre-commit hooks, add them to the `.pre-commit-config.yaml` file.

awx_zipline_ai-0.0.32.dist-info/RECORD

@@ -0,0 +1,96 @@
+__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agent/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+agent/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+agent/ttypes.py,sha256=264pSOgpRPm6XMjdwxdi9W7ciTYElPby7smK7z18RBQ,64304
+ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/airflow_helpers.py,sha256=MVkSIrpBMx3XqvAORxcuiwaLdQDijV1yFT5j1PSkNWw,8760
+ai/chronon/constants.py,sha256=74zhArZB-h9jWcNe41kQr2GEC5qgcSw_V9yIwwW0O_E,165
+ai/chronon/group_by.py,sha256=ubrbJrKyYqZsJY2O4UMKXpHanNgcD789lHl-S3XXnjU,27735
+ai/chronon/join.py,sha256=VM8qG1_IDwXprBtLjm3zSZJjQ57MZPlpXue5dJdjOjs,25817
+ai/chronon/logger.py,sha256=rPyeV8uAFvJRY57vJg9KFyEjE4XMl8nVYEoHN1y2dmY,861
+ai/chronon/model.py,sha256=UVF0_-NF2Zdy5zLHXDibFzKMfuMfmYCcOK80cnT7vzA,1115
+ai/chronon/query.py,sha256=AUgUwVeGWH2DKldMa2gt34DiZI3GUM3PtnY26DYgeNs,5417
+ai/chronon/source.py,sha256=xl3Gb925893nSEFuFkWXpuk10kQySyR3XePMNpf8WMs,4016
+ai/chronon/staging_query.py,sha256=Fv4BLEhYjx-YYZbMmBhwlQBTPfEXgUNLBZpTkj9qBvU,8519
+ai/chronon/types.py,sha256=ppH8i3PYVicOQE3h5DWsvtEFTTVnoEAcW8tn7Htc89E,1501
+ai/chronon/utils.py,sha256=D1idc9Y2YYb3Qf2K-CQhh4kGm5Glc0vRSe141y89DPs,15701
+ai/chronon/windows.py,sha256=gvtp1-XHu9yp9xkOy4oDaTGDn7GfIa0JhFXb_RBm12Q,1333
+ai/chronon/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/cli/git_utils.py,sha256=6uez-TOvuz9rsUcyq9JYTITvwJy9I-JVZAyX2CNdki8,4255
+ai/chronon/cli/logger.py,sha256=2VVleE_tt4E9ysLHRR3bckfSWEhFugjwOIVB6KYFWDE,1437
+ai/chronon/cli/compile/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/cli/compile/column_hashing.py,sha256=9aMCJo0ITt4GUNStajo4evkU2zGmH8XtDOJJuLqGvjg,13760
+ai/chronon/cli/compile/compile_context.py,sha256=DFINBU2l5ISenPlP2rSme1-tRd5svYVM8l0iOEv1nQ0,6648
+ai/chronon/cli/compile/compiler.py,sha256=GUQblOz-fMCZ4b-VAiS1FGocjnlBH7nexw7X_Py2TXY,6913
+ai/chronon/cli/compile/conf_validator.py,sha256=ezzvKeJykvFTzCWoMZF8E4qUUFRkGLCgTBW_ofXGYgY,29058
+ai/chronon/cli/compile/fill_templates.py,sha256=LHZ6xZMyplm1rqr56ziiZpyu5RJOhNtsbLrKC-kyPg4,1204
+ai/chronon/cli/compile/parse_configs.py,sha256=xiaM4fX3Ubxpjya6MphMHd045bMu8fyjZ8zbeTo2DWk,4292
+ai/chronon/cli/compile/parse_teams.py,sha256=x8LyHJGcbv_iDSqyvev-ll4mo96KXpfoOfb8ND7rmDY,7962
+ai/chronon/cli/compile/serializer.py,sha256=Qky6AhrRTxhyNSKXD5_Jp6VecNUdmGy32tCL8BtKBY4,4272
+ai/chronon/cli/compile/version_utils.py,sha256=1T6HamqMkfk-zel4E0-0BIn5I82mumC98FimFu_zjgs,1220
+ai/chronon/cli/compile/display/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/cli/compile/display/class_tracker.py,sha256=QEn2pySXDMaNTFfigZW2aPRyYLF38vrru9yZ1QohlW4,3671
+ai/chronon/cli/compile/display/compile_status.py,sha256=rvQ3CecKbIx_3VabFcJnwZyhLK_Bo5GEoSDXGSe1Ca4,3094
+ai/chronon/cli/compile/display/compiled_obj.py,sha256=hsstubchqQDBcFmarRLd4DgMt_sUvTmDlwmNBP01-VI,217
+ai/chronon/cli/compile/display/console.py,sha256=UpMqeJ0C8i0pkue1AHnnyyX0bFJ9zZeJ7HBR6yhuA8A,54
+ai/chronon/cli/compile/display/diff_result.py,sha256=VOkCqImR2TrhUmuWea4GDo3VqkMnVHmbmoLmR-6uS6U,4120
+ai/chronon/repo/__init__.py,sha256=HQ60TFAvCI9uO164uBggIOMs99KPWd_1gae_mMJIfgQ,1413
+ai/chronon/repo/aws.py,sha256=iaFaE-J31Du4hNzx9vknOX5j-A-1OfzeGBhstO2FogE,11745
+ai/chronon/repo/cluster.py,sha256=_dUFMPmjgzRQSSKJVP2qwyGfQLPw0BiUnlUNRPqJY1c,4972
+ai/chronon/repo/compile.py,sha256=bJCDe0Ubv74eOkVDfSeJ3BMcb6yI03Wy4jvzQBivQH4,1820
+ai/chronon/repo/constants.py,sha256=JRDjvV2ZOSIxWJfXmnBngvtRIDBiU8A7869yTzZDiJc,5075
+ai/chronon/repo/default_runner.py,sha256=u2mJrw1xZc8ECK_v9gQSshcVrJDmBb8L9oo7UnuoMAQ,11180
+ai/chronon/repo/explore.py,sha256=rmpUwODc69Gdvx9rYj8vx-Xg5UWD9k1TvBDrs716oUE,14584
+ai/chronon/repo/extract_objects.py,sha256=cyXX07icGTHUzrw__SeQpO76rPMt7jHFnEW08lz56io,4761
+ai/chronon/repo/gcp.py,sha256=z4fTxe0D8z22O52888XvooL-yxMGhNq8aBcTw5Jr1-A,25117
+ai/chronon/repo/gitpython_utils.py,sha256=HIcQNFMf68G0qHvIbk7HSwkcqFA97YrcMsgXw5qw-ow,435
+ai/chronon/repo/hub_runner.py,sha256=jU3x3cwbsClWvmg2eyOMCDLOV6RhbIedKs89IAgP5Wc,8920
+ai/chronon/repo/hub_uploader.py,sha256=2oAAnIru1d7mDq5xAYRt7KWlZ0o_YhQujfGpsL5ph9E,3995
+ai/chronon/repo/init.py,sha256=9Xhf4oeKuDtxpiS6kNf6rOju8-u7VyQDy3BTiMoSaS4,1740
+ai/chronon/repo/join_backfill.py,sha256=WLN9TY6bpupd5T5BFcrTH6qSqpy39P8O9ZvrEhs9vzQ,4024
+ai/chronon/repo/run.py,sha256=Kex79ebLH0bDdq1k1dBK1EC7Y8691trMsvzrTYRQtVs,10780
+ai/chronon/repo/serializer.py,sha256=-nLxTIT-YR7evDCCTAsL_8NusrRz-wFwn-AcBFF7Ocw,4979
+ai/chronon/repo/team_json_utils.py,sha256=JsUc_KwTFn4VMn1ZYwJAPTIeSOfUsDoywzTsBr9Oi70,1472
+ai/chronon/repo/utils.py,sha256=iKJt-qzV3GNLo8yTnc-X7NzKIpweozpS777NddCA74Q,19148
+ai/chronon/repo/zipline.py,sha256=ktWaSbvto3FK5LmB5lsbztOlhcYEbNeVGmznPMk8R3w,1050
+ai/chronon/repo/zipline_hub.py,sha256=laNoRdu_WrAbwE62YMCb7S2pxV4J5JZ0laKW4Za7mTs,12889
+ai/chronon/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/teams.py,sha256=UBSSZjvSYf4MnujqJgyxmDUbjX8KZS8HBoPyn29SH9o,2505
+ai/chronon/resources/gcp/group_bys/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/group_bys/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/group_bys/test/data.py,sha256=UG6_q9aKXiGFX2P4BOSIL6UIQjT-4wHpWXl96UaQcnk,1089
+ai/chronon/resources/gcp/joins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/joins/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/joins/test/data.py,sha256=mhDpA_zKHiHNwpsku8AqsNL-WDA_LH81CO405tIZ9Og,784
+ai/chronon/resources/gcp/sources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/sources/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai/chronon/resources/gcp/sources/test/data.py,sha256=Im-tN2ReJ_K1syRX6IXKoICNwv45tvynBgqTL0qkluQ,1092
+gen_thrift/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gen_thrift/api/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+gen_thrift/api/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+gen_thrift/api/ttypes.py,sha256=1AGnaSImuXHAVdL6IldrQmBfBW-5Or-D64BgIT7EfHU,147829
+gen_thrift/common/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+gen_thrift/common/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+gen_thrift/common/ttypes.py,sha256=zvxBcnY6OPaioA3LepUgQIxaHCsNTuPUuxbtxwjR9u8,58939
+gen_thrift/eval/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+gen_thrift/eval/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+gen_thrift/eval/ttypes.py,sha256=uky--UKz-2vuN6Z7QN7ctEqoyPx69GsfOBZAtIdN79M,28560
+gen_thrift/fetcher/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+gen_thrift/fetcher/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+gen_thrift/fetcher/ttypes.py,sha256=XdTAL7ql_bQQT4M_eJqs9yOecE6gqa6gO5RoCwsRGWE,4553
+gen_thrift/hub/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+gen_thrift/hub/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+gen_thrift/hub/ttypes.py,sha256=tn1VJ3y-sS3NN_gRttTHWmnNhdbSGb4rypApT4NCdBQ,39453
+gen_thrift/observability/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+gen_thrift/observability/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+gen_thrift/observability/ttypes.py,sha256=QTKvQ9zqgbj4UcujIs8LuvDXvuyONWTrtsoiQB6ULxE,82254
+gen_thrift/planner/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
+gen_thrift/planner/constants.py,sha256=Z7iGhv-3ePR_U5Z70F0Cqycz6WBJbGvMaJ80bwMaIjc,388
+gen_thrift/planner/ttypes.py,sha256=EXfaw3bPwax8Bl4EfF9b-5KwQW0-9kby4NCiZnIIZZ8,67348
+awx_zipline_ai-0.0.32.dist-info/METADATA,sha256=98q1QZCGm-OjXIHGXhu85sieN8XUBNpCaCkZIV3mDf4,5790
+awx_zipline_ai-0.0.32.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+awx_zipline_ai-0.0.32.dist-info/entry_points.txt,sha256=EZGwU6w_Ve4BZ2z8zYrIJW0zMMon0IK2FIvIhoSzeqE,60
+awx_zipline_ai-0.0.32.dist-info/top_level.txt,sha256=PY-p-EGZt3eyJqEK5pzRx38U921A1_aRWyCS0a0blvQ,29
+awx_zipline_ai-0.0.32.dist-info/RECORD,,

awx_zipline_ai-0.0.32.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

awx_zipline_ai-0.0.32.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+zipline = ai.chronon.repo.zipline:zipline

awx_zipline_ai-0.0.32.dist-info/top_level.txt

@@ -0,0 +1,4 @@
+__init__
+agent
+ai
+gen_thrift

gen_thrift/api/__init__.py

@@ -0,0 +1 @@
+__all__ = ['ttypes', 'constants']

gen_thrift/api/constants.py

@@ -0,0 +1,15 @@
+#
+# Autogenerated by Thrift Compiler (0.22.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+#  options string: py
+#
+
+from thrift.Thrift import TType, TMessageType, TFrozenDict, TException, TApplicationException
+from thrift.protocol.TProtocol import TProtocolException
+from thrift.TRecursive import fix_spec
+from uuid import UUID
+
+import sys
+from .ttypes import *