Inconsistent results with 'sort', 'cache', and AQE.

import spark.implicits._
val df = (1 to 4).toDF("id").sort("id").sample(0.4, 123)

println("NON CACHED:")

println("  count: " + df.count())
println("  collect: " + df.collect().mkString(" "))
df.show()

println("CACHED:")
df.cache().count()

println("  count: " + df.count())
println("  collect: " + df.collect().mkString(" "))
df.show()

df.unpersist()

NON CACHED:
  count: 2
  collect: [1] [4]
+---+
| id|
+---+
|  1|
|  4|
+---+

CACHED:
  count: 3
  collect: [1] [4]
+---+
| id|
+---+
|  1|
|  2|
|  3|
+---+