class AnimalsScreenViewModel(
    private val someRepository: SomeRepository,
    @DefaultDispatcher private val defaultDispatcher: CoroutineDispatcher,
) : ViewModel() {

    private val _animalType: MutableStateFlow<AnimalType> = MutableStateFlow(<http://AnimalType.Cat|AnimalType.Cat>)
    val animalType: StateFlow<AnimalType>
        get() = _animalType.asStateFlow()

    val animalListStateFlow: StateFlow<List<Animal>> = animalType
        .map { animalType: AnimalType ->
            coroutineScope {
                val getAnimalAsyncCalls: List<Deferred<List<Animal>>> = List(4) { index ->
                    async {
                        someRepository.getAnimals(
                            animalType = animalType,
                            page = (index + 1)
                        )
                    }
                }
                val animalList: List<Animal> = getAnimalAsyncCalls
                    .awaitAll()
                    .flatten()
                animalList
            }
        }
        .stateIn(
            scope = viewModelScope + defaultDispatcher, // Not sure about setting a dispatcher here
            started = SharingStarted.WhileSubscribed(5000),
            initialValue = listOf()
        )

    fun setAnimalType(animalType: AnimalType) {
        _animalType.value = animalType
    }
}

.map { animalType: AnimalType ->
    coroutineScope {
        val getAnimalAsyncCalls: List<Deferred<List<Animal>>> = List(4) { index ->
            async {
                petFinderRepository.getAnimals(
                    animalType = animalType,
                    page = (index + 1)
                )
            }
        }
        val animalList: List<Animal> = getAnimalAsyncCalls
            .awaitAll()
            .flatten()
        animalList
    }
}

animalType

transform