Auto fixes from PEP8, fixes from flake8.

2024-05-15 15:18:17 +02:00
parent 9f526a8f3b
commit 4bcd28d0ea
15 changed files with 391 additions and 417 deletions
@@ -1,8 +1,5 @@
 import pytest
-import os
-from unittest import mock
-from scraibe import diarisation, Diariser
-
+from scraibe import Diariser


@pytest.fixture
@@ -15,11 +12,10 @@ def diariser_instance():
    Returns:
        Diariser(Obj): An instance of the Diariser class with a mocked token.
    """
-    #with mock.patch.object(Diariser, '_get_token', return_value = 'HF_TOKEN' ):
+    # with mock.patch.object(Diariser, '_get_token', return_value = 'HF_TOKEN' ):
    return Diariser('pyannote')


-
 def test_Diariser_init(diariser_instance):
    """Test the initialization of the Diariser class.

@@ -30,18 +26,7 @@ def test_Diariser_init(diariser_instance):
    Args:
        diariser_instance (obj): instance of the Diariser class

-    Returns: 
+    Returns:
           None
-    """    
+    """
    assert diariser_instance.model == 'pyannote'
-
-
-
-
-
-
-
-
-
-
-