Commit 30e30945 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent aa530dca
Loading
Loading
Loading
Loading
+7 −2
Original line number Diff line number Diff line
@@ -173,12 +173,17 @@ class GeneralL1DAG(BaseDAG):
        if not demo:
            # push and update
            res_push = self.push_dag_group_run(dag_group_run, dag_run_list)
            print(f"{len(dag_run_list)} DAG runs -> {res_push}")
            print(
                f"{len(dag_run_list)} DAG runs -> "
                f"{json.dumps(dag_group_run, indent=None, separators=(',', ':'))} -> "
                f"{res_push}"
            )
            assert res_push.success, res_push.message
        else:
            # no push
            print(
                f"{len(dag_run_list)} DAG runs -> {json.dumps(dag_group_run, indent=None, separators=(',', ':'))}"
                f"{len(dag_run_list)} DAG runs -> "
                f"{json.dumps(dag_group_run, indent=None, separators=(',', ':'))}"
            )
        # TODO: `dag_group_run` and `dag_run_list` should be dumped to a text file in the future
        return dag_group_run, dag_run_list