From 6a288cd30eb6b86263f876454e09abe2d6df1449 Mon Sep 17 00:00:00 2001
From: CaptainOfHacks <39195263+CaptainOfHacks@users.noreply.github.com>
Date: Mon, 18 Sep 2023 16:53:25 +0300
Subject: [PATCH] Update test_dag_a.py

---
 dags/test_dag_a.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/dags/test_dag_a.py b/dags/test_dag_a.py
index c5e4183b..dfdd358f 100644
--- a/dags/test_dag_a.py
+++ b/dags/test_dag_a.py
@@ -2,6 +2,7 @@
 
 from airflow.decorators import dag, task
 from airflow.models.param import Param
+from airflow.utils.types import NOTSET
 
 from dags import DEFAULT_DAG_ARGUMENTS
 from dags.dags_utils import get_dag_param
@@ -17,20 +18,21 @@
      tags=['test', 'new-ui'],
      params={
          FORM_NUMBER_DAG_PARAM: Param(
-             type="string",
+             default=NOTSET,
+             type=["null", "string"],
              title="Form number",
              description="""Form number of the notice"""),
          START_DATE_DAG_PARAM: Param(
-             f"{datetime.date.today()}",
-             type="string",
+             default=f"{datetime.date.today()}",
+             type=["null", "string"],
              format="date",
              title="Date Picker",
              description="Please select a date, use the button on the left for a pup-up calendar. "
                          "See that here are no times!",
          ),
          END_DATE_DAG_PARAM: Param(
-             f"{datetime.date.today()}",
-             type="string",
+             default=f"{datetime.date.today()}",
+             type=["null", "string"],
              format="date",
              title="Date Picker",
              description="Please select a date, use the button on the left for a pup-up calendar. "