We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 90880e2 commit e6e2bcfCopy full SHA for e6e2bcf
pytensor/link/pytorch/linker.py
@@ -51,7 +51,6 @@ class wrapper:
51
"""
52
53
def __init__(self, fn, gen_functors):
54
- self._fn = fn
55
self.fn = torch.compile(fn)
56
self.gen_functors = gen_functors.copy()
57
pytensor/link/utils.py
@@ -758,9 +758,7 @@ def fgraph_to_python(
758
new_output_name = unique_name(i)
759
getter_unique_name = unique_name(getter_or_value)
760
global_env[getter_unique_name] = getter_or_value
761
- assign_str = (
762
- f"{new_output_name} = {getter_unique_name}()"
763
- )
+ assign_str = f"{new_output_name} = {getter_unique_name}()"
764
body_assigns.append(assign_str)
765
node_input_names.append(new_output_name)
766
continue
0 commit comments