|
25 | 25 |
|
26 | 26 |
|
27 | 27 | with DAG(
|
28 |
| - dag_id="sunroof.solar_potential_by_census_tract", |
| 28 | + dag_id="sunroof_solar.solar_potential_by_censustract", |
29 | 29 | default_args=default_args,
|
30 | 30 | max_active_runs=1,
|
31 | 31 | schedule_interval="@daily",
|
|
36 | 36 | # Run CSV transform within kubernetes pod
|
37 | 37 | transform_csv = kubernetes_pod.KubernetesPodOperator(
|
38 | 38 | task_id="transform_csv",
|
39 |
| - name="solar_potential_by_census_tract", |
| 39 | + name="solar_potential_by_censustract", |
40 | 40 | namespace="default",
|
41 | 41 | affinity={
|
42 | 42 | "nodeAffinity": {
|
|
56 | 56 | }
|
57 | 57 | },
|
58 | 58 | image_pull_policy="Always",
|
59 |
| - image="{{ var.json.sunroof.container_registry.run_csv_transform_kub }}", |
| 59 | + image="{{ var.json.sunroof_solar.container_registry.run_csv_transform_kub }}", |
60 | 60 | env_vars={
|
61 | 61 | "SOURCE_URL": "gs://project-sunroof/csv/latest/project-sunroof-census_tract.csv",
|
62 | 62 | "SOURCE_FILE": "files/data.csv",
|
63 | 63 | "TARGET_FILE": "files/data_output.csv",
|
64 | 64 | "CHUNKSIZE": "750000",
|
65 | 65 | "TARGET_GCS_BUCKET": "{{ var.value.composer_bucket }}",
|
66 |
| - "TARGET_GCS_PATH": "data/sunroof/solar_potential_by_census_tract/data_output.csv", |
| 66 | + "TARGET_GCS_PATH": "data/sunroof/solar_potential_by_censustract/data_output.csv", |
67 | 67 | },
|
68 | 68 | resources={"limit_memory": "8G", "limit_cpu": "3"},
|
69 | 69 | )
|
|
72 | 72 | load_to_bq = gcs_to_bigquery.GCSToBigQueryOperator(
|
73 | 73 | task_id="load_to_bq",
|
74 | 74 | bucket="{{ var.value.composer_bucket }}",
|
75 |
| - source_objects=["data/sunroof/solar_potential_by_census_tract/data_output.csv"], |
| 75 | + source_objects=["data/sunroof/solar_potential_by_censustract/data_output.csv"], |
76 | 76 | source_format="CSV",
|
77 |
| - destination_project_dataset_table="sunroof.solar_potential_by_census_tract", |
| 77 | + destination_project_dataset_table="sunroof.solar_potential_by_censustract", |
78 | 78 | skip_leading_rows=1,
|
79 | 79 | allow_quoted_newlines=True,
|
80 | 80 | write_disposition="WRITE_TRUNCATE",
|
|
0 commit comments