import os
from pathlib import Path

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

import tensorflow as tf

tf.get_logger().setLevel("ERROR")
tf.keras.utils.set_random_seed(7)

CHECKPOINT_DIR = Path("training_checkpoints")
FEATURES = tf.constant(
    [[0.2, 0.8], [0.9, 0.1], [0.1, 0.7], [0.8, 0.2]],
    dtype=tf.float32,
)
LABELS = tf.constant([[0.0], [1.0], [0.0], [1.0]], dtype=tf.float32)
PROBE = tf.constant([[0.15, 0.75], [0.85, 0.25]], dtype=tf.float32)