Fix invalid wam package imports

data/testing_data/impulses/create_impulses.py

@@ -3,7 +3,7 @@
 import numpy as np
 from pydub import AudioSegment
 from pydub.generators import Sine
-from wam import IMPULSE_DIR, CODA_DIR
+from wham import IMPULSE_DIR, CODA_DIR
 
 MIN_NUM_CLICKS = 4
 MAX_NUM_CLICKS = 8

wham/embedding/__init__.py

@@ -1,4 +1,4 @@
-from wam import VAMPNET_DIR
+from wham import VAMPNET_DIR
 from pathlib import Path
 
 VAMPNET_MDL_PATH = VAMPNET_DIR / "models"

wham/embedding/generate_embeddings.py

@@ -1,8 +1,9 @@
 from collections import Counter
 from pathlib import Path
 import pandas as pd
-from wam import VAMPNET_DIR, EMBEDDING_PLOTS_DIR, SNIPPETS_DIR, DENOISED_DATA_DIR, CSV_PATH
-from wam.utils import pathify
+from codatools.codatools import SNIPPETS_DIR
+from wham import VAMPNET_DIR, EMBEDDING_PLOTS_DIR, DENOISED_DATA_DIR, CSV_PATH
+from wham.utils import pathify
 import argparse
 import logging
 from shutil import rmtree

wham/embedding/models.py

@@ -9,7 +9,7 @@
 from preprocessing import preprocess_audio
 from audiotools import AudioSignal
 from vampnet.interface import Interface
-from wam import VAMPNET_DIR
+from wham import VAMPNET_DIR
 import wandb
 from pathlib import Path
 

wham/generation/eval/calculate_custom_fad.py

@@ -1,7 +1,7 @@
 import os
 import numpy as np
 from scipy.linalg import sqrtm
-from wam import CODA_EMBEDDING_DIR, EVALUATE_EMBEDDING_DIR
+from wham import CODA_EMBEDDING_DIR, EVALUATE_EMBEDDING_DIR
 import argparse
 
 """

wham/generation/eval/mse_error_by_freq.py

@@ -5,7 +5,7 @@
 import scipy.signal as signal
 from collections import defaultdict
 
-from wam import CODA_DIR, REGEN_CODA_DIR
+from wham import CODA_DIR, REGEN_CODA_DIR
 
 TARGET_SAMPLERATE = 44100
 

wham/generation/generate_output_snippets.py

@@ -11,10 +11,10 @@
 from vampnet.interface import Interface
 from vampnet import mask as pmask
 from prompt_configs import CONFIGS
-from wam.utils import move_wav_files
+from wham.utils import move_wav_files
 import torch
 
-from wam import GENERATED_MARINE_PATH, GENERATED_MARINE_PATH_TEMP, MARINE_TEST_DIR, VAMPNET_DIR
+from wham import GENERATED_MARINE_PATH, GENERATED_MARINE_PATH_TEMP, MARINE_TEST_DIR, VAMPNET_DIR
 
 import os
 

wham/generation/tokenizer_only_generation.py

@@ -13,9 +13,9 @@
 from vampnet.interface import Interface
 from vampnet import mask as pmask
 from prompt_configs import CONFIGS
-from wam.utils import move_wav_files
+from wham.utils import move_wav_files
 
-from wam import GENERATED_MARINE_PATH_TEMP, MARINE_TEST_DIR, VAMPNET_DIR, REGEN_CODA_DIR
+from wham import GENERATED_MARINE_PATH_TEMP, MARINE_TEST_DIR, VAMPNET_DIR, REGEN_CODA_DIR
 
 import os