import tensorflow as tf | |
from training_pipeline.models import model | |
class ModelTest(tf.test.TestCase): | |
def testBuildKerasModel(self): | |
pass | |
# built_model = model._build_keras_model( | |
# ["foo", "bar"] | |
# ) # pylint: disable=protected-access | |
# self.assertEqual(len(built_model.inputs), 2) | |
if __name__ == "__main__": | |
tf.test.main() | |