I'm also not so surprised it got canceled. As I hinted at above, it seems the writers couldn't get it to go where they wanted without handing out the idiot ball to characters left and right or just suddenly change their personality for forced reasons.
It's too bad, there was some good writing in there, some interesting ideas to explore, and season 2 wasn't enough time to explore them all, but they tried to cram it in there anyway.
My writer's room headcanon is now that they had outlines for five seasons, realized they couldn't get them, tried to fit two seasons in this one and hoping they would at least get one season more for a counterinsurgency.
The final twist was fun, with the allegiance of her partner and her superior. But also not really thought through. Part of me wants to rewatch S02 knowing the twist, but it's not good enough to rewatch.