Issue in CompressibleFluid CFD_Hydra.py

I encountered an issue with running CompressibleFluid CFD_Hydra.py
There is an error on utils.py in CompressibleFluid

u = u.loc[0].set(d0)
      ^^^^^
AttributeError: 'jaxlib.xla_extension.ArrayImpl' object has no attribute 'loc

From my perspective, jnp object shouldn't be using "loc" since it cannot work with pandas dataframe functions
Any ways to solve this?