How to infer type of select for subquery?

const subquery: SUBQUERY = tx
        .select({ id: this.testTable.id })
        .from(this.testTable)
        .where(eq(this.testTable.launchId, launchId))

  async findByTestIds(testIds: SUBQUERY) {
    return await this.db
      .select()
      .from(this.testArgumentsTable)
      .where(inArray(this.testArgumentsTable.testId, testIds));
  }