I have a 2D array of 1's and 0's that I wanna use as a sensor, what's the best way?

I am building a 2D tile grid game and I want the Agent to be able to see the board state.
I tried a grid sensor but it keeps telling me the shape of the grid is wrong and I read that it is a bug that can be fixed by updating ML Agents, but I am already using 3.0.0-exp.1, the same shape error comes up when I use the camera sensor.
So, I feel like 2D array is my best bet.

I have a 2D array of 1’s and 0’s that I wanna use as a sensor, I currently have it like this:


I am using AddOneHotObservation, and I feel like this seems wrong is the sensor really seeing it as a 2D grid? I feel like it’s not.

Any insight would be much appreciated.