Fix AdvancedSubtensor static shape with newaxis

Author: ricardoV94

pytensor/tensor/subtensor.py

@@ -2630,7 +2630,7 @@ def make_node(self, x, *indices):
         adv_group_axis = None
         last_adv_group_axis = None
         expanded_x_shape = tuple(
-            np.insert(np.array(x.type.shape, dtype=object), 1, new_axes)
+            np.insert(np.array(x.type.shape, dtype=object), new_axes, values=1)
         )
         for i, (idx, dim_length) in enumerate(
             zip_longest(explicit_indices, expanded_x_shape, fillvalue=NoneSliceConst)

tests/tensor/test_subtensor.py

@@ -1854,6 +1854,7 @@ def test_static_shape(self):
 
         assert x[idx1].type.shape == (10, None)
         assert x[:, idx1].type.shape == (None, 10)
+        assert x[None, :, idx1].type.shape == (1, None, 10)
         assert x[idx2, :5].type.shape == (3, None, None)
         assert specify_shape(x, (None, 7))[idx2, :5].type.shape == (3, None, 5)
         assert specify_shape(x, (None, 3))[idx2, :5].type.shape == (3, None, 3)