neural-amp-modeler

commit 78813fa9fd57a4d60619c8bde8b89e77ed839fdf
parent e61ebb5b285b667d40232e071f35b488e296d8aa
Author: Steven Atkinson <[email protected]>
Date:   Sat,  9 Sep 2023 18:52:31 -0700

Add pedal_amp to GearType enum (#307)


Diffstat:
M
bin/train/easy_colab.ipynb
 | 
2
+-
M
nam/models/metadata.py
 | 
1
+

2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/bin/train/easy_colab.ipynb b/bin/train/easy_colab.ipynb
@@ -114,7 +114,7 @@
         "gear_model = \"GearAmp\" #@param {type:\"string\"}\n",
         "#@markdown Gear type:\n",
         "# This needs to be a literal. You need to change it by hand if you change the enum!\n",
-        "gear_type = \"Pick from: amp, pedal, amp_cab, amp_pedal_cab, preamp, studio\" #@param {type:\"string\"}\n",
+        "gear_type = f\"Pick from: amp, pedal, pedal_amp, amp_cab, amp_pedal_cab, preamp, studio\" #@param {type:\"string\"}\n",
         "#@markdown Tone type:\n",
         "tone_type = \"Pick from: clean, overdrive, crunch, hi_gain, fuzz\" #@param {type:\"string\"}\n",
         "\n",
diff --git a/nam/models/metadata.py b/nam/models/metadata.py
@@ -18,6 +18,7 @@ __all__ = ["GearType", "ToneType", "Date", "UserMetadata"]
 class GearType(Enum):
     AMP = "amp"
     PEDAL = "pedal"
+    PEDAL_AMP = "pedal_amp"
     AMP_CAB = "amp_cab"
     AMP_PEDAL_CAB = "amp_pedal_cab"
     PREAMP = "preamp"