Fix Build Dependencies. Pin dependencies for more stable support of VST training.

jesseengel · Magenta Team · commit 761d61a5f133 · 2023-05-24T17:42:46.000-07:00
PiperOrigin-RevId: 535042575
diff --git a/ddsp/training/metrics.py b/ddsp/training/metrics.py
@@ -29,6 +29,8 @@
 MIN_F0_CONFIDENCE = 0.85
 OUTLIER_MIDI_THRESH = 12
 
+# pytype: disable=signature-mismatch  # overriding-parameter-count-checks
+
 
 # ---------------------- Helper Functions --------------------------------------
 def squeeze(input_vector):
@@ -181,7 +183,7 @@ def __init__(self, sample_rate, frame_rate, name='loudness'):
   def metrics(self):
     return self._metrics
 
-  def update_state(self, batch, audio_gen):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
+  def update_state(self, batch, audio_gen):
     """Update metrics based on a batch of audio.
 
     Args:
@@ -228,7 +230,7 @@ def __init__(self, sample_rate, frame_rate, name='f0_crepe'):
   def metrics(self):
     return self._metrics
 
-  def update_state(self, batch, audio_gen):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
+  def update_state(self, batch, audio_gen):
     """Update metrics based on a batch of audio.
 
     Args:
@@ -293,7 +295,7 @@ def __init__(
   def metrics(self):
     return self._metrics
 
-  def update_state(self, batch, f0_hz_predict):  # pytype: disable=signature-mismatch  # overriding-parameter-count-checks
+  def update_state(self, batch, f0_hz_predict):
     """Update metrics based on a batch of audio.
 
     Args:
diff --git a/ddsp/version.py b/ddsp/version.py
@@ -18,4 +18,4 @@
 pulling in all the dependencies in __init__.py.
 """
 
-__version__ = '3.6.0'
+__version__ = '3.7.0'
diff --git a/pylintrc b/pylintrc
@@ -169,12 +169,6 @@ disable=abstract-method,
 # mypackage.mymodule.MyReporterClass.
 output-format=text
 
-# Put messages in a separate file for each module / package specified on the
-# command line instead of printing them on stdout. Reports (if any) will be
-# written in a file name "pylint_global.[txt|html]". This option is deprecated
-# and it will be removed in Pylint 2.0.
-files-output=no
-
 # Tells whether to display a full report or only the messages
 reports=no
 
@@ -293,12 +287,6 @@ ignore-long-lines=(?x)(
 # else.
 single-line-if-stmt=yes
 
-# List of optional constructs for which whitespace checking is disabled. `dict-
-# separator` is used to allow tabulation in dicts, etc.: {1  : 1,\n222: 2}.
-# `trailing-comma` allows a space between comma and closing bracket: (a, ).
-# `empty-line` allows space-only lines.
-no-space-check=
-
 # Maximum number of lines in a module
 max-module-lines=99999
 
diff --git a/setup.py b/setup.py
@@ -38,26 +38,26 @@
     scripts=[],
     install_requires=[
         'absl-py',
-        'cloudml-hypertune',
+        'cloudml-hypertune<=0.1.0.dev6',
         'crepe<=0.0.12',
         'dill<=0.3.4',
         'future',
         'gin-config>=0.3.0',
         'google-cloud-storage',
         'hmmlearn<=0.2.7',
-        'librosa',
-        'pydub',
+        'librosa<=0.10',
+        'pydub<=0.25.1',
         'protobuf<=3.20',  # temporary fix for proto dependency bug
-        'mir_eval',
+        'mir_eval<=0.7',
         'note_seq<0.0.4',
         'numpy<1.24',
-        'scipy',
+        'scipy<=1.10.1',
         'six',
-        'tensorflow',
+        'tensorflow<=2.11',
         'tensorflowjs<3.19',
-        'tensorflow-probability',
-        'tensorflow-datasets',
-        'tflite_support'
+        'tensorflow-probability<=0.19',
+        'tensorflow-datasets<=4.9',
+        'tflite_support<=0.1'
     ],
     extras_require={
         'gcp': [
