Skip to content

Commit

Permalink
changing structure to be similiar to smfp
Browse files Browse the repository at this point in the history
  • Loading branch information
patriciacatandi committed Apr 17, 2024
1 parent 39b6d9a commit c1b64a7
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions pipelines/templates/run_dbt_model/flows.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,22 +15,22 @@

from pipelines.constants import constants

templates__run_dbt_model_smas__flow = deepcopy(templates__run_dbt_model__flow)
templates__run_dbt_model_smas__flow.state_handlers = [handler_inject_bd_credentials]
templates__run_dbt_model_smi__flow = deepcopy(templates__run_dbt_model__flow)
templates__run_dbt_model_smi__flow.state_handlers = [handler_inject_bd_credentials]

templates__run_dbt_model_smas__flow.storage = GCS(constants.GCS_FLOWS_BUCKET.value)
templates__run_dbt_model_smas__flow.run_config = KubernetesRun(
templates__run_dbt_model_smi__flow.storage = GCS(constants.GCS_FLOWS_BUCKET.value)
templates__run_dbt_model_smi__flow.run_config = KubernetesRun(
image=constants.DOCKER_IMAGE.value,
labels=[
constants.RJ_SMFP_AGENT_LABEL.value,
constants.RJ_SMI_AGENT_LABEL.value,
],
)

templates_run_dbt_model_smas_default_parameters = {
templates_run_dbt_model_smi_default_parameters = {
"dataset_id": "dataset_id",
"table_id": "table_id",
}
templates__run_dbt_model_smas__flow = set_default_parameters(
templates__run_dbt_model_smas__flow,
default_parameters=templates_run_dbt_model_smas_default_parameters,
templates__run_dbt_model_smi__flow = set_default_parameters(
templates__run_dbt_model_smi__flow,
default_parameters=templates_run_dbt_model_smi_default_parameters,
)

0 comments on commit c1b64a7

Please sign in to comment.