Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Preserve path to the file when rendering model #3186

Merged
merged 5 commits into from
Mar 3, 2023
Merged

Preserve path to the file when rendering model #3186

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
a1d5aab
Issue #3184
LysSanzMoreta Feb 28, 2023
da67516
reverted signature
LysSanzMoreta Mar 2, 2023
02c4254
os import
LysSanzMoreta Mar 2, 2023
fbbd6d6
black formatting
LysSanzMoreta Mar 2, 2023
109467e
split line
LysSanzMoreta Mar 3, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions pyro/infer/inspect.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
# SPDX-License-Identifier: Apache-2.0

import itertools
import os
from collections import defaultdict
from pathlib import Path
from types import SimpleNamespace
Expand Down Expand Up @@ -572,7 +573,7 @@ def render_model(
list of tuples for semisupervised models.
:param model_kwargs: Dict of keyword arguments to pass to the model, or
list of dicts for semisupervised models.
:param str filename: File to save rendered model in.
:param str filename: Name of file or path to file to save rendered model in.
:param bool render_distributions: Whether to include RV distribution
annotations (and param constraints) in the plot.
:param bool render_params: Whether to show params inthe plot.
Expand Down Expand Up @@ -604,9 +605,9 @@ def render_model(
graph = render_graph(graph_spec, render_distributions=render_distributions)

if filename is not None:
filename = Path(filename)
suffix = filename.suffix[1:] # remove leading period from suffix
graph.render(filename.stem, view=False, cleanup=True, format=suffix)
suffix = Path(filename).suffix[1:] # remove leading period from suffix
filepath = os.path.splitext(filename)[0]
graph.render(filepath, view=False, cleanup=True, format=suffix)

return graph

Expand Down