 
     // `inBatches` is lazy so no elements were accessed.
     XCTAssert(
-      dataset.accessed.allSatisfy(){ !$0 },
+      dataset.accessed.allSatisfy { !$0 },
       "No elements should have been accessed yet.")
     for batches in epochs {
       for (i, batch) in batches.enumerated() {


    // `inBatches` is lazy so no elements were accessed.
    XCTAssert(
      dataset.accessed.allSatisfy(){ !$0 },
      "No elements should have been accessed yet.")
    for (i, batch) in batches.enumerated() {
      // Elements are not accessed until we do something with `batch` so only
      // the elements up to `i * batchSize` have been accessed yet.
      XCTAssert(
        dataset.accessed[..<(i * batchSize)].allSatisfy(){ $0 },
        "Not all elements prior to \(i * batchSize) have been accessed.")
      XCTAssert(
        dataset.accessed[(i * batchSize)...].allSatisfy(){ !$0 },
        "Some elements after \(i * batchSize) have been accessed.")
      let _ = Array(batch)
      let limit = (i + 1) * batchSize
      // We accessed elements up to `limit` but no further.
      XCTAssert(
        dataset.accessed[..<limit].allSatisfy(){ $0 },
        "Not all elements prior to \(limit) have been accessed.")
      XCTAssert(
        dataset.accessed[limit...].allSatisfy(){ !$0 },
        "Some elements after \(limit) have been accessed.")
    }
  }

      for k in 0..<64 {
        let currentShape = nonuniformDataset[i * 64 + k].shape[0]
        let paddedPart =
          atStart ? b[k, 0..<(expectedShape - currentShape)] : (b[k, currentShape..<expectedShape])
        XCTAssertEqual(
          paddedPart,
          Tensor<Int32>(
