Remove type annotations

oarriaga · Apr 15, 2024 · 0300eef · 0300eef
1 parent 0dc3483
commit 0300eef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paz/pipelines/detection.py b/paz/pipelines/detection.py
@@ -924,8 +924,8 @@ class DetectVVAD(Processor):
         average_type: String. 'mean' or 'weighted'. How the predictions are averaged. Set averaging_window_size to 1 to
             disable averaging
     """
-    def __init__(self, architecture: Architecture_Options = 'CNN2Plus1D_Light', stride=10, averaging_window_size=6,
-                 average_type: Average_Options = 'weighted', offsets=[0, 0], colors=[[0, 255, 0], [255, 0, 0]]):
+    def __init__(self, architecture='CNN2Plus1D_Light', stride=10, averaging_window_size=6,
+                 average_type='weighted', offsets=[0, 0], colors=[[0, 255, 0], [255, 0, 0]]):
         super(DetectVVAD, self).__init__()
         self.offsets = offsets
         self.colors = colors