[BUG] Possible naming error

[norhan-synnada]

Bug Report

Describe the Bug

When below model is compiled, a bug occurs in PhysicalModel (possibly in naming). Same name is assigned to different outputs.

To Reproduce

model = Model()

to_list = ToList(n=3)
concat_1 = Concat()
concat_2 = Concat()

input1 = IOKey("input1", type=Tensor)
input2 = IOKey("input2", type=Tensor)
input3 = IOKey("input3", type=Tensor)

model |= to_list(input1=input1, input2=input2, input3=input3)
model += concat_1(output="output_1")
indexed_data = to_list.output[-2:]
model |= concat_2(input=indexed_data, output="output_2")

Expected Behavior

Should be compiled with no errors.

System Info

• OS:
• Mithril Version:
• Torch Version:
• MLX Version:
• NumPy Version:
• JAX Version:
• Python Version:
• Architecture:

CC: @kberat-synnada, @norhan-synnada