neural-amp-modeler

commit 952b98d8b5b846102f69490d056f7b017fbc842e
parent 039e858eae4e5daf129bc2c3c3c540e07d36cc01
Author: Steven Atkinson <steven@atkinson.mn>
Date:   Sat, 11 Feb 2023 18:58:02 -0800

Pass delay_interpolation_method in init_from_config

Diffstat:
M
nam/data.py
 | 
7
+++++++

1 file changed, 7 insertions(+), 0 deletions(-)
diff --git a/nam/data.py b/nam/data.py
@@ -201,6 +201,10 @@ class Dataset(AbstractDataset, InitializableFromConfig):
             completely dry signal (i.e. connecting the interface output directly back
             into the input with which the guitar was originally recorded.)
         """
+        if not isinstance(delay_interpolation_method, _DelayInterpolationMethod):
+            delay_interpolation_method = _DelayInterpolationMethod(
+                delay_interpolation_method
+            )
         x, y = [z[start:stop] for z in (x, y)]
         if delay is not None and delay != 0:
             x, y = self._apply_delay(x, y, delay, delay_interpolation_method)
@@ -266,6 +270,9 @@ class Dataset(AbstractDataset, InitializableFromConfig):
             "start": config.get("start"),
             "stop": config.get("stop"),
             "delay": config.get("delay"),
+            "delay_interpolation_method": config.get(
+                "delay_interpolation_method", _DelayInterpolationMethod.CUBIC.value
+            ),
             "y_scale": config.get("y_scale", 1.0),
             "x_path": config["x_path"],
             "y_path": config["y_path"],