[-] importlib-resources is not needed

This commit is contained in:
2023-10-11 18:54:27 -04:00
parent 21c7b38cd4
commit d47289931d
2 changed files with 3 additions and 4 deletions
+2 -2
View File
@@ -1,3 +1,4 @@
from pathlib import Path
import warnings
warnings.filterwarnings("ignore")
import random
@@ -12,7 +13,6 @@ import math
import torchaudio.transforms as T
import os
import torch
from importlib_resources import files
import argparse
import yaml
import sys
@@ -41,7 +41,7 @@ class CLAPWrapper():
def get_config_path(self, version):
if version in self.supported_versions:
return files('configs').joinpath(f"config_{version}.yml").read_text()
return (Path(__file__).parent / f"configs/config_{version}.yml").read_text()
else:
raise ValueError(f"The specific version is not supported. The supported versions are {str(self.supported_versions)}")