Merge branch 'fix/update-dense-weight-id' into 'main'

Update dense weight id See merge request es/ai/hannah/hannah!381
ekut-es · Apr 18, 2024 · cbd819c · cbd819c
2 parents f7bf3a3 + 8b49fce
commit cbd819c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hannah/nas/functional_operators/operators.py b/hannah/nas/functional_operators/operators.py
@@ -187,6 +187,7 @@ def __call__(self, *operands) -> Any:
         new_linear = super().__call__(*operands)
         new_linear.in_features = operands[1].shape()[0]
         new_linear.out_features = operands[1].shape()[1]
+        operands[1].id = f"{new_linear.id}.{operands[1].id}"
         return new_linear
 
     def shape_fun(self):