fix: correctly check completion status in run_dag magic (#3878)

TobikoData · Feb 20, 2025 · d566301 · d566301
1 parent 1e84ba5
commit d566301
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sqlmesh/magics.py b/sqlmesh/magics.py
@@ -493,7 +493,7 @@ def run_dag(self, context: Context, line: str) -> None:
         """Evaluate the DAG of models using the built-in scheduler."""
         args = parse_argstring(self.run_dag, line)
 
-        success = context.run(
+        completion_status = context.run(
             args.environment,
             start=args.start,
             end=args.end,
@@ -503,7 +503,7 @@ def run_dag(self, context: Context, line: str) -> None:
             exit_on_env_update=args.exit_on_env_update,
             no_auto_upstream=args.no_auto_upstream,
         )
-        if not success:
+        if completion_status.is_failure:
             raise SQLMeshError("Error Running DAG. Check logs for details.")
 
     @magic_arguments()