From f030aee457ffb5e1b94f9566c42c5a81412d27dc Mon Sep 17 00:00:00 2001 From: Feras Saad Date: Sun, 12 Apr 2026 18:49:37 -0400 Subject: [PATCH] Fix #130: sppl_to_graph ignores the provided file extension --- magics/magics.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/magics/magics.py b/magics/magics.py index d70a7ec..695df6d 100644 --- a/magics/magics.py +++ b/magics/magics.py @@ -49,14 +49,18 @@ def sppl_to_python(self, line): def sppl_to_graph(self, line, local_ns): tokens = line.strip().split(' ') line = tokens[0] - filename = tokens[1] if len(tokens) == 2 else None + if 2 <= len(tokens): + filename = tokens[1].split('.')[0] + ext = tokens[1].split('.')[-1] + else: + (filename, ext) = (None, None) if line in self.programs: spe = self.sppl_get_spe(line) elif line in local_ns: spe = local_ns[line] else: assert False, 'unknown program %s' % (line,) - return render_graphviz(spe, filename=filename) + return render_graphviz(spe, filename=filename, ext=ext) @line_magic def sppl_get_namespace(self, line):