add sg

Co-authored-by: Thomas Chow <[email protected]>

Author: tchow-zlai

api/py/test/sample/group_bys/quickstart/purchases.py

@@ -15,7 +15,6 @@
 from ai.chronon.api.ttypes import Source, EventSource
 from ai.chronon.query import Query, select
 from ai.chronon.group_by import GroupBy, Aggregation, Operation
-from ai.chronon.api.ttypes import StagingQuery, MetaData
 
 """
 This GroupBy aggregates metrics about a user's previous purchases in various windows.
@@ -68,25 +67,3 @@
         ),
     ],
 )
-
-
-query = """
-SELECT
-    ts,
-    ds,
-    purchase_id,
-    user_id,
-    product_id,
-    purchase_price
-FROM data.purchases
-WHERE ds BETWEEN '{{ start_date }}' AND '{{ end_date }}'
-"""
-
-staging_query = StagingQuery(
-    query=query,
-    startPartition="2023-10-31",
-    metaData=MetaData(
-        name='purchases_staging_query',
-        outputNamespace="data"
-    )
-)

api/py/test/sample/staging_queries/quickstart/purchases.py

@@ -0,0 +1,38 @@
+#     Copyright (C) 2023 The Chronon Authors.
+#
+#     Licensed under the Apache License, Version 2.0 (the "License");
+#     you may not use this file except in compliance with the License.
+#     You may obtain a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#     Unless required by applicable law or agreed to in writing, software
+#     distributed under the License is distributed on an "AS IS" BASIS,
+#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#     See the License for the specific language governing permissions and
+#     limitations under the License.
+
+from ai.chronon.api.ttypes import StagingQuery, MetaData
+
+
+
+query = """
+SELECT
+    ts,
+    ds,
+    purchase_id,
+    user_id,
+    product_id,
+    purchase_price
+FROM data.purchases
+WHERE ds BETWEEN '{{ start_date }}' AND '{{ end_date }}'
+"""
+
+staging_query = StagingQuery(
+    query=query,
+    startPartition="2023-10-31",
+    metaData=MetaData(
+        name='purchases_staging_query',
+        outputNamespace="data"
+    )
+)