import airflow
from project_config import project

from conductor.core import Conductor

default_args = {"start_date": airflow.utils.dates.days_ago(0)}

c = Conductor(
    "my-dag", project=project, schedule_interval="@once", default_args=default_args
)
dag = c.dag


# Test Processing
def test_function(a, b):
    print(a, b)


process_op = c.operators.SageMakerProcessingOperator(
    task_id="process", entrypoint=test_function, a=1, b="two"
)
