Fix flaky HighAvailabilityTaskAssignorIntegrationTest.shouldScaleOutWithWarmupTasksAndPersistentStores

[2024-03-13T16:07:11.125Z] Gradle Test Run :streams:test > Gradle Test Executor 95 > HighAvailabilityTaskAssignorIntegrationTest > shouldScaleOutWithWarmupTasksAndPersistentStores(String, TestInfo) > "shouldScaleOutWithWarmupTasksAndPersistentStores(String, TestInfo).balance_subtopology" FAILED
[2024-03-13T16:07:11.125Z] java.lang.AssertionError: the first assignment after adding a node should be unstable while we warm up the state.
[2024-03-13T16:07:11.125Z] at org.apache.kafka.streams.integration.HighAvailabilityTaskAssignorIntegrationTest.assertFalseNoRetry(HighAvailabilityTaskAssignorIntegrationTest.java:310)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.streams.integration.HighAvailabilityTaskAssignorIntegrationTest.lambda$shouldScaleOutWithWarmupTasks$7(HighAvailabilityTaskAssignorIntegrationTest.java:237)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.test.TestUtils.lambda$waitForCondition$3(TestUtils.java:395)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.test.TestUtils.retryOnExceptionWithTimeout(TestUtils.java:443)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.test.TestUtils.waitForCondition(TestUtils.java:392)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.test.TestUtils.waitForCondition(TestUtils.java:376)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.test.TestUtils.waitForCondition(TestUtils.java:366)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.streams.integration.HighAvailabilityTaskAssignorIntegrationTest.shouldScaleOutWithWarmupTasks(HighAvailabilityTaskAssignorIntegrationTest.java:232)
[2024-03-13T16:07:11.125Z] at org.apache.kafka.streams.integration.HighAvailabilityTaskAssignorIntegrationTest.shouldScaleOutWithWarmupTasksAndPersistentStores(HighAvailabilityTaskAssignorIntegrationTest.java:130)

https://ci-builds.apache.org/blue/rest/organizations/jenkins/pipelines/Kafka/pipelines/kafka-pr/branches/PR-15474/runs/12/nodes/9/steps/88/log/?start=0