flake8 changes

This commit is contained in:
shahules786 2022-10-05 15:47:25 +05:30
parent 670b70938a
commit babe5776ce
4 changed files with 8 additions and 14 deletions

View File

@ -1,6 +1,5 @@
import glob import glob
import os import os
from re import S
import numpy as np import numpy as np
from scipy.io import wavfile from scipy.io import wavfile

View File

@ -119,7 +119,8 @@ class Avergeloss(nn.Module):
def validate_loss(self, loss: str): def validate_loss(self, loss: str):
if loss not in LOSS_MAP.keys(): if loss not in LOSS_MAP.keys():
raise ValueError( raise ValueError(
f"Invalid loss function {loss}, available loss functions are {tuple([loss for loss in LOSS_MAP.keys()])}" f"""Invalid loss function {loss}, available loss functions are
{tuple([loss for loss in LOSS_MAP.keys()])}"""
) )
else: else:
return LOSS_MAP[loss] return LOSS_MAP[loss]

View File

@ -1,16 +1,10 @@
try:
from functools import cached_property
except ImportError:
from backports.cached_property import cached_property
from importlib import import_module from importlib import import_module
from huggingface_hub import cached_download, hf_hub_url from huggingface_hub import cached_download, hf_hub_url
import logging
import numpy as np import numpy as np
import os import os
from typing import Optional, Union, List, Text, Dict, Any from typing import Optional, Union, List, Text, Dict, Any
from torch.optim import Adam from torch.optim import Adam
import torch import torch
from torch.nn.functional import pad
import pytorch_lightning as pl import pytorch_lightning as pl
from pytorch_lightning.utilities.cloud_io import load as pl_load from pytorch_lightning.utilities.cloud_io import load as pl_load
from urllib.parse import urlparse from urllib.parse import urlparse
@ -19,7 +13,6 @@ from pathlib import Path
from enhancer import __version__ from enhancer import __version__
from enhancer.data.dataset import EnhancerDataset from enhancer.data.dataset import EnhancerDataset
from enhancer.utils.io import Audio
from enhancer.loss import Avergeloss from enhancer.loss import Avergeloss
from enhancer.inference import Inference from enhancer.inference import Inference

View File

@ -19,6 +19,7 @@ def check_files(root_dir: str, files: Files):
def merge_dict(default_dict: dict, custom: Optional[dict] = None): def merge_dict(default_dict: dict, custom: Optional[dict] = None):
params = dict(default_dict) params = dict(default_dict)
if custom: if custom:
params.update(custom) params.update(custom)