remove u()

holgern · Feb 28, 2025 · 76a05f9 · 76a05f9
1 parent d1b9383
commit 76a05f9
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/pyedflib/edfwriter.py b/pyedflib/edfwriter.py
@@ -988,7 +988,9 @@ def writeSamples(self, data_list: Union[List[np.ndarray], np.ndarray], digital:
                 if success<0:
                     raise OSError(f'Unknown error while calling writeSamples: {success}')
 
-    def writeAnnotation(self, onset_in_seconds: Union[int, float], duration_in_seconds: Union[int, float], description: str, str_format: str = 'utf_8') -> int:
+    def writeAnnotation(self, onset_in_seconds: Union[int, float],
+                        duration_in_seconds: Union[int, float],
+                        description: str, str_format: str = 'utf_8') -> int:
         """
         Writes an annotation/event to the file
         """
@@ -1000,16 +1002,26 @@ def writeAnnotation(self, onset_in_seconds: Union[int, float], duration_in_secon
 
         if str_format == 'utf_8':
             if duration_in_seconds >= 0:
-                return write_annotation_utf8(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), np.round(duration_in_seconds*10000).astype(int), du(description))
+                return write_annotation_utf8(self.handle,
+                                             np.round(onset_in_seconds*10000).astype(np.int64),
+                                             np.round(duration_in_seconds*10000).astype(int),
+                                             du(description))
             else:
-                return write_annotation_utf8(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), -1, du(description))
+                return write_annotation_utf8(self.handle,
+                                             np.round(onset_in_seconds*10000).astype(np.int64),
+                                             -1, du(description))
         else:
             if duration_in_seconds >= 0:
                 # FIX: description must be bytes. string will fail in u function
-                return write_annotation_latin1(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), np.round(duration_in_seconds*10000).astype(int), u(description).encode('latin1'))  # type: ignore
+                return write_annotation_latin1(self.handle,
+                                               np.round(onset_in_seconds*10000).astype(np.int64),
+                                               np.round(duration_in_seconds*10000).astype(int), description.encode('latin1'))  # type: ignore
             else:
                 # FIX: description must be bytes. string will fail in u function
-                return write_annotation_latin1(self.handle, np.round(onset_in_seconds*10000).astype(np.int64), -1, u(description).encode('latin1'))  # type: ignore
+                return write_annotation_latin1(self.handle,
+                                               np.round(onset_in_seconds*10000).astype(np.int64),
+                                               -1,
+                                               description.encode('latin1'))  # type: ignore
 
     def close(self) -> None:
         """