From 68ca2ba1551948895c8cb0060d56cf7ca59bf1a8 Mon Sep 17 00:00:00 2001
From: Simon Sorg <simon.sorg@student.hpi.de>
Date: Mon, 30 Oct 2023 14:51:40 +0100
Subject: [PATCH] fix: remove imports

---
 src/diart/models.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/diart/models.py b/src/diart/models.py
index be774b44..7df9ac09 100644
--- a/src/diart/models.py
+++ b/src/diart/models.py
@@ -1,15 +1,13 @@
 from abc import ABC, abstractmethod
-from typing import Optional, Text, Union, Callable, Mapping, TYPE_CHECKING
+from typing import Optional, Text, Union, Callable, TYPE_CHECKING
 
 import torch
 import torch.nn as nn
 from requests import HTTPError
 
 try:
-    import pyannote.audio.pipelines.utils as pyannote_loader
     from pyannote.audio import Inference, Model
     from pyannote.audio.pipelines.speaker_verification import (
-        WeSpeakerPretrainedSpeakerEmbedding,
         PretrainedSpeakerEmbedding,
     )
 
@@ -18,8 +16,9 @@
     _has_pyannote = False
 
 if TYPE_CHECKING:
+    from pyannote.audio import Model
     from pyannote.audio.pipelines.speaker_verification import (
-        WeSpeakerPretrainedSpeakerEmbedding,
+        PretrainedSpeakerEmbedding,
     )