Skip to content

Commit

Permalink
Merge pull request #67 from KevKibe/main
Browse files Browse the repository at this point in the history
Update: Ignore warnings
  • Loading branch information
KevKibe authored Apr 4, 2024
2 parents b32cc95 + a3e8c4a commit 47fa666
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/training/audio_data_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
from transformers import PreTrainedTokenizer
from typing import Dict, Any
import librosa
import warnings
warnings.filterwarnings("ignore")

class AudioDataProcessor:
"""
Expand Down
2 changes: 2 additions & 0 deletions src/training/collator.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
import torch
from dataclasses import dataclass
from typing import Any, Dict, List, Union
import warnings

warnings.filterwarnings("ignore")

@dataclass
class DataCollatorSpeechSeq2SeqWithPadding:
Expand Down
2 changes: 2 additions & 0 deletions src/training/evaluation.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
import evaluate
import warnings

warnings.filterwarnings("ignore")

class MetricComputer:
"""
Expand Down
2 changes: 2 additions & 0 deletions src/training/load_data.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
from datasets import load_dataset, DatasetDict, IterableDataset
import warnings

warnings.filterwarnings("ignore")

class Dataset:
"""
Expand Down

0 comments on commit 47fa666

Please sign in to comment.