Fix colab error (#17)

EleutherAI · Mar 9, 2024 · 12d249b · 12d249b
1 parent d25a7fa
commit 12d249b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/amt/audio.py b/amt/audio.py
@@ -265,6 +265,8 @@ def __init__(
         )
 
     def _get_paths(self, dir_path):
+        os.makedirs(dir_path, exist_ok=True)
+
         return [
             os.path.join(dir_path, f)
             for f in os.listdir(dir_path)