Dimension error for Hierarchical model

Questions version agnostic
10

@ricardoV94 This is my error I get when I tried your proposed solution

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
File c:\Users\pymc_hier\.venv\lib\site-packages\aesara\link\vm.py:414, in Loop.__call__(self)
    411 for thunk, node, old_storage in zip_longest(
    412     self.thunks, self.nodes, self.post_thunk_clear, fillvalue=()
    413 ):
--> 414     thunk()
    415     for old_s in old_storage:

File c:\Users\pymc_hier\.venv\lib\site-packages\aesara\graph\op.py:543, in Op.make_py_thunk..rval(p, i, o, n, params)
    539 @is_thunk_type
    540 def rval(
    541     p=p, i=node_input_storage, o=node_output_storage, n=node, params=None
    542 ):
--> 543     r = p(n, [x[0] for x in i], o)
    544     for o in node.outputs:

File c:\Users\pymc_hier\.venv\lib\site-packages\aesara\tensor\blas.py:1997, in Dot22Scalar.perform(self, node, inp, out)
   1996 try:
-> 1997     z[0] = np.asarray(scalar * np.dot(x, y))
   1998 except ValueError as e:
   1999     # The error raised by numpy has no shape information, we
   2000     # mean to add that

File <__array_function__ internals>:200, in dot(*args, **kwargs)
...
 |TensorConstant{[[15.  5. ..  4.  3.]]} [id K] 
 |TensorConstant{-1.0} [id L]