import os

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

import tensorflow as tf

tf.get_logger().setLevel("ERROR")


def make_batch(rows):
    return tf.ones((rows, 3), dtype=tf.float32)