Update inspect.py

pyro-ppl · May 5, 2022 · 17d2488 · 17d2488
1 parent 558563c
commit 17d2488
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pyro/infer/inspect.py b/pyro/infer/inspect.py
@@ -552,6 +552,9 @@ def render_model(
     assert model_args is None or isinstance(
         model_args, tuple
     ), "model_args must be None or tuple"
+    assert model_kwargs is None or isinstance(
+        model_kwargs, dict
+    ), "model_kwargs must be None or dict"
     relations = get_model_relations(model, model_args, model_kwargs)
     graph_spec = generate_graph_specification(relations, render_params=render_params)
     graph = render_graph(graph_spec, render_distributions=render_distributions)