fix type error in prepare_tfrecord_lib.py

PiperOrigin-RevId: 677825559
magenta · Sep 23, 2024 · a2d0517 · a2d0517
1 parent 72f543d
commit a2d0517
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/ddsp/training/data_preparation/prepare_tfrecord_lib.py b/ddsp/training/data_preparation/prepare_tfrecord_lib.py
@@ -175,7 +175,7 @@ def prepare_tfrecord(input_audio_paths,
                      chunk_secs=20.0,
                      center=False,
                      viterbi=True,
-                     pipeline_options=''):
+                     pipeline_options=()):
   """Prepares a TFRecord for use in training, evaluation, and prediction.
 
   Args:

diff --git a/setup.py b/setup.py
@@ -38,6 +38,7 @@
     scripts=[],
     install_requires=[
         'absl-py',
+        'apache-beam',
         'cloudml-hypertune<=0.1.0.dev6',
         'crepe<=0.0.12',
         'dill<=0.3.4',