Fixing failing AWS integration tests (#654)

## Summary

I think this is the fix for the failing CI test -- we now require this
to be set to create airflow deps.

## Checklist
- [ ] Added Unit Tests
- [ ] Covered by existing CI
- [ ] Integration tested
- [ ] Documentation update



<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **Bug Fixes**
- Improved error handling when accessing the partition column
configuration, preventing unexpected errors if the configuration is
missing.

- **Tests**
- Added a default partition column configuration for teams to enhance
test coverage.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Co-authored-by: ezvz <[email protected]>

Author: varant-zlai

api/python/ai/chronon/airflow_helpers.py

@@ -73,8 +73,11 @@ def _get_airflow_deps_from_source(source, partition_column=None):
 
 
 def extract_default_partition_column(obj):
-    return obj.metaData.executionInfo.conf.common.get("spark.chronon.partition.column")
-
+    try:
+        return obj.metaData.executionInfo.conf.common.get("spark.chronon.partition.column")
+    except Exception:
+        # Error handling occurs in `create_airflow_dependency`
+        return None
 
 def _set_join_deps(join):
     default_partition_col = extract_default_partition_column(join)

api/python/test/canary/teams.py

@@ -6,6 +6,11 @@
     description="Default team",
     email="ml-infra@<customer>.com",  # TODO: Infra team email
     outputNamespace="default",
+    conf=ConfigProperties(
+        common={
+            "spark.chronon.partition.column": "ds",
+        }
+    ),
     env=EnvironmentVariables(
         common={
             "VERSION": "latest",

api/thrift/agent.thrift

@@ -1,4 +1,5 @@
 namespace java ai.chronon.api
+include "common.thrift"
 
 // TODO: Need to brainstorm and make necessary changes. just a starting point to unblock other work.
 struct YarnAutoScalingSpec {
@@ -33,7 +34,7 @@ struct YarnJob {
     1: optional string appName
     2: optional YarnJobType jobType
 
-    10: optional list<string> args
+    10: optional list<string> argsList
     11: optional map<string, string> env
     12: optional map<string, string> conf
     // creates local file with this name and contents - relative to cwd
@@ -121,13 +122,9 @@ struct JobInfo {
     10: optional YarnIncrementalJobStatus yarnIncrementalStatus
 }
 
-struct DatePartitionRange {
-    1: optional string start
-    2: optional string end
-}
 
 struct PartitionListingPutRequest {
-    1: optional map<PartitionListingJob, list<DatePartitionRange>> partitions
+    1: optional map<PartitionListingJob, list<common.DateRange>> partitions
     2: optional map<PartitionListingJob, string> errors
 }