awx-zipline-ai 0.3.3__py3-none-any.whl → 0.3.5__py3-none-any.whl

ai/chronon/repo/default_runner.py

@@ -27,6 +27,7 @@ class Runner:
         self.online_jar = args.get(ONLINE_JAR_ARG)
         self.online_class = args.get(ONLINE_CLASS_ARG)
         self.online_args = args.get(ONLINE_ARGS)
+        self.warehouse_bucket = args.get("warehouse_bucket", utils.get_default_warehouse_bucket())
 
         self.conf_type = (args.get("conf_type") or "").replace(
             "-", "_"

ai/chronon/repo/gcp.py

@@ -17,7 +17,6 @@ from ai.chronon.repo.utils import (
     check_call,
     check_output,
     extract_filename_from_path,
-    get_customer_warehouse_bucket,
     get_environ_arg,
     retry_decorator,
     split_date_range,
@@ -248,7 +247,7 @@ class GcpRunner(Runner):
             )
             gcs_files.append(
                 GcpRunner.upload_gcs_blob(
-                    get_customer_warehouse_bucket(), metadata_conf_path, destination_file_path
+                    self.warehouse_bucket, metadata_conf_path, destination_file_path
                 )
             )
 

ai/chronon/repo/run.py

@@ -78,6 +78,7 @@ def set_defaults(ctx):
         "render_info": os.path.join(chronon_repo_path, RENDER_INFO_DEFAULT_SCRIPT),
         "project_conf": obj.get("project_conf"),
         "artifact_prefix": os.environ.get("ARTIFACT_PREFIX"),
+        "warehouse_bucket": os.environ.get("WAREHOUSE_BUCKET"),
         "flink_state_uri": os.environ.get("FLINK_STATE_URI"),
     }
     for key, value in defaults.items():
@@ -202,6 +203,10 @@ def validate_additional_jars(ctx, param, value):
     "--artifact-prefix",
     help="Remote artifact URI to install zipline client artifacts necessary for interacting with Zipline infrastructure.",
 )
+@click.option(
+    "--warehouse-bucket",
+    help="GCS or S3 bucket to use as warehouse for staging metadata/configs.",
+)
 @click.option("--disable-cloud-logging", is_flag=True, default=False, help="Disables cloud logging")
 @click.option(
     "--enable-debug",
@@ -250,6 +255,7 @@ def main(
     validate_rows,
     join_part_name,
     artifact_prefix,
+    warehouse_bucket,
     disable_cloud_logging,
     additional_jars,
     enable_debug,

ai/chronon/repo/utils.py

@@ -57,7 +57,7 @@ def get_environ_arg(env_name, ignoreError=False) -> str:
     return value
 
 
-def get_customer_warehouse_bucket() -> str:
+def get_default_warehouse_bucket() -> str:
     return f"zipline-warehouse-{get_customer_id()}"
 
 

{awx_zipline_ai-0.3.3.dist-info → awx_zipline_ai-0.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awx-zipline-ai
-Version: 0.3.3
+Version: 0.3.5
 Summary: CLI tool for the Zipline AI platform
 Author-email: Zipline AI <hello@zipline.ai>
 License: Apache License 2.0

{awx_zipline_ai-0.3.3.dist-info → awx_zipline_ai-0.3.5.dist-info}/RECORD

@@ -40,19 +40,19 @@ 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/default_runner.py,sha256=aD4eBfODIM-gl8XZ3fAP7y2-3jzN7dCMriEiPeiCDDQ,11279
 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/gcp.py,sha256=hFlmIZ7-mfaGmhjsdmmAhmSLfpvr9aDJoHNxKzrXWXg,25072
 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/run.py,sha256=v91ZEL4lZ5pi2mOX2qdfAE5C_xsXFVKiUR2ZgB_VWx4,10988
 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/utils.py,sha256=vuwxMNbC1UMzecdOVkwH-i1_IlZUzVOgjl4u3YDT09w,19147
 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
@@ -89,8 +89,8 @@ gen_thrift/observability/ttypes.py,sha256=QTKvQ9zqgbj4UcujIs8LuvDXvuyONWTrtsoiQB
 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.3.3.dist-info/METADATA,sha256=ipOB-T85OAXMkrrpSbQD78Fn4PapWIQjjuJv5wQ_fek,5726
-awx_zipline_ai-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-awx_zipline_ai-0.3.3.dist-info/entry_points.txt,sha256=EZGwU6w_Ve4BZ2z8zYrIJW0zMMon0IK2FIvIhoSzeqE,60
-awx_zipline_ai-0.3.3.dist-info/top_level.txt,sha256=PY-p-EGZt3eyJqEK5pzRx38U921A1_aRWyCS0a0blvQ,29
-awx_zipline_ai-0.3.3.dist-info/RECORD,,
+awx_zipline_ai-0.3.5.dist-info/METADATA,sha256=VbjG35aLStavwwkmyCHd-rtqjbzdVHAmCb9WvBn9QQE,5726
+awx_zipline_ai-0.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+awx_zipline_ai-0.3.5.dist-info/entry_points.txt,sha256=EZGwU6w_Ve4BZ2z8zYrIJW0zMMon0IK2FIvIhoSzeqE,60
+awx_zipline_ai-0.3.5.dist-info/top_level.txt,sha256=PY-p-EGZt3eyJqEK5pzRx38U921A1_aRWyCS0a0blvQ,29
+awx_zipline_ai-0.3.5.dist-info/RECORD,,