It is, in fairness, a pretty common trope. See also https://www.smbc-comics.com/index.php?db=comics&id=936 , and also Iain M Banks' Surface Detail; both predate _The Good Place_.
(WARNING: The above comic contains the same spoiler as the article, more or less.)
Jacob's Ladder (1990) is one I'm thinking of too.