Internal change.

PiperOrigin-RevId: 326259440
magenta · Aug 12, 2020 · 31b42c2 · 31b42c2
1 parent 91375fe
commit 31b42c2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ddsp/losses_test.py b/ddsp/losses_test.py
@@ -33,8 +33,8 @@ def test_output_shape_is_correct(self):
         loudness_weight=1.0,
     )
 
-    input_audio = tf.random.uniform((3, 8000), dtype=tf.float32)
-    target_audio = tf.random.uniform((3, 8000), dtype=tf.float32)
+    input_audio = tf.ones((3, 8000), dtype=tf.float32)
+    target_audio = tf.ones((3, 8000), dtype=tf.float32)
 
     loss = loss_obj(input_audio, target_audio)
 
@@ -47,8 +47,8 @@ class PretrainedCREPEEmbeddingLossTest(tf.test.TestCase):
   def test_output_shape_is_correct(self):
     loss_obj = losses.PretrainedCREPEEmbeddingLoss()
 
-    input_audio = tf.random.uniform((3, 16000), dtype=tf.float32)
-    target_audio = tf.random.uniform((3, 16000), dtype=tf.float32)
+    input_audio = tf.ones((3, 16000), dtype=tf.float32)
+    target_audio = tf.ones((3, 16000), dtype=tf.float32)
 
     loss = loss_obj(input_audio, target_audio)