make mypy happy

atvaccaro · atvaccaro · commit b57ae3d26160 · 2023-05-22T13:07:32.000-04:00
diff --git a/warehouse/scripts/visualize.py b/warehouse/scripts/visualize.py
@@ -283,11 +283,13 @@ def ci_report(
             latest_dir,
         ]
     ).result
+    assert isinstance(modified_models, list)
+    assert isinstance(changed_or_downstream_incremental_models, list)
     include = set(changed_or_downstream_incremental_models + modified_models)
     typer.secho(f"Visualizing the following models: {include}")
     viz(
         "man",
-        include=list(include),  # this is List typed
+        include=list(include),
         output=Path("./target/dag.png"),
     )
 

Original file line number	Diff line number	Diff line change
`@@ -283,11 +283,13 @@ def ci_report(`
`283`	`283`	`latest_dir,`
`284`	`284`	`]`
`285`	`285`	`).result`
	`286`	`+ assert isinstance(modified_models, list)`
	`287`	`+ assert isinstance(changed_or_downstream_incremental_models, list)`
`286`	`288`	`include = set(changed_or_downstream_incremental_models + modified_models)`
`287`	`289`	`typer.secho(f"Visualizing the following models: {include}")`
`288`	`290`	`viz(`
`289`	`291`	`"man",`
`290`		`- include=list(include), # this is List typed`
	`292`	`+ include=list(include),`
`291`	`293`	`output=Path("./target/dag.png"),`
`292`	`294`	`)`
`293`	`295`