list->tuple in output shape

NNPDF · Mar 4, 2024 · b2cd6fb · b2cd6fb
1 parent 8899f31
commit b2cd6fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/n3fit/src/n3fit/backends/keras_backend/multi_dense.py b/n3fit/src/n3fit/backends/keras_backend/multi_dense.py
@@ -125,7 +125,7 @@ def compute_output_shape(self, input_shape):
         output_shape = super().compute_output_shape(input_shape)
 
         # Add back the replica axis to the output shape.
-        output_shape = output_shape[:1] + [self.replicas] + output_shape[1:]
+        output_shape = output_shape[:1] + (self.replicas,) + output_shape[1:]
 
         return output_shape