Skip to content

Commit b4582d9

Browse files
add spark partition column
Co-authored-by: Thomas Chow <[email protected]>
1 parent 6e96ab2 commit b4582d9

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

api/python/test/sample/teams.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,16 @@
11
from ai.chronon.api.ttypes import Team
22
from ai.chronon.repo.constants import RunMode
3-
from ai.chronon.types import EnvironmentVariables
3+
from ai.chronon.types import EnvironmentVariables, ConfigProperties
44

55
default = Team(
66
description="Default team",
77
email="ml-infra@<customer>.com", # TODO: Infra team email
88
outputNamespace="default",
9+
conf=ConfigProperties(
10+
common={
11+
"spark.chronon.partition.column": "_DATE",
12+
}
13+
),
914
env=EnvironmentVariables(
1015
common={
1116
"VERSION": "latest",
@@ -43,6 +48,11 @@
4348

4449
test = Team(
4550
outputNamespace="test",
51+
conf=ConfigProperties(
52+
common={
53+
"spark.chronon.partition.column": "_test_column",
54+
}
55+
),
4656
env=EnvironmentVariables(
4757
common={
4858
"GCP_BIGTABLE_INSTANCE_ID": "test-instance" # example, custom bigtable instance

0 commit comments

Comments
 (0)