   }
 
   public func sample<G: RandomNumberGenerator>(generator: inout G) -> ShapedArray<Int32> {
-    return ShapedArray<Int32>(Int32.random(in: 0..<size, using: &generator))
+    return ShapedArray(Int32.random(in: 0..<size, using: &generator))
   }
 
   public func contains(_ value: ShapedArray<Int32>) -> Bool {

  }
}

public struct Box<Scalar: TensorFlowScalar & Comparable & Hashable & Codable>: Space {
  public let low: ShapedArray<Scalar>
  public let high: ShapedArray<Scalar>
  public let shape: [Int]

    for _ in 0..<size {
      scalars.append(Int32.random(in: 0...1, using: &generator))
    }
    return ShapedArray(shape: [size], scalars: scalars)
  }

  public func contains(_ value: ShapedArray<Int32>) -> Bool {
