Skip to content

Commit

Permalink
feat: handling wrong model type
Browse files Browse the repository at this point in the history
  • Loading branch information
SamuelLarkin committed Oct 8, 2024
1 parent 3336f9b commit 94af73d
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion dfaligner/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
import torch.nn as nn
from everyvoice.model.aligner.config import AlignerConfig
from everyvoice.text.text_processor import TextProcessor
from everyvoice.utils import pydantic_validation_error_shortener

from .config import DFAlignerConfig
from .duration_extraction import extract_durations_with_dijkstra
Expand Down Expand Up @@ -42,7 +43,17 @@ def __init__(
) -> None:
super().__init__()
if isinstance(config, dict):
config = AlignerConfig(**config)
from pydantic import ValidationError

try:
config = AlignerConfig(**config)
except ValidationError as e:
from loguru import logger

logger.error(f"{pydantic_validation_error_shortener(e)}")
raise TypeError(
"Unable to load config. Possible causes: is it really a AlignerConfig? or the correct version?"
)
self.config: AlignerConfig = config # type: ignore
self.preprocessed_dir = Path(self.config.preprocessing.save_dir)
self.sep = "--"
Expand Down

0 comments on commit 94af73d

Please sign in to comment.