From d84a8ae48916f47fd5156c3d1af1e314640b63b5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20De=20Keersmaeker?=
 <francois.dekeersmaeker@uclouvain.be>
Date: Thu, 7 Nov 2024 21:53:12 +0000
Subject: [PATCH] Fixed type hints

---
 profile_translator_blocklist/translator.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/profile_translator_blocklist/translator.py b/profile_translator_blocklist/translator.py
index 5499764..28880b9 100644
--- a/profile_translator_blocklist/translator.py
+++ b/profile_translator_blocklist/translator.py
@@ -5,7 +5,7 @@ of NFTables firewall script and NFQueue C source code.
 
 ## Imports
 # Libraries
-from typing import Iterator
+from typing import Iterable
 import os
 import importlib
 import yaml
@@ -301,7 +301,7 @@ def translate_policy(
 
 def translate_policies(
         device:       dict,
-        policies:     Iterator[dict],
+        policies:     Iterable[dict],
         nfqueue_name: str     = None,
         nfqueue_id:   int     = 0,
         output_dir:   str     = os.getcwd(),
@@ -316,7 +316,7 @@ def translate_policies(
 
     Args:
         device (dict): Device metadata
-        policies (Iterator[dict]): iterator over policies
+        policies (Iterable[dict]): iterable containing the policies to translate
         nfqueue_id (int): NFQueue start index for this profile's policies (must be an integer between 0 and 65535)
         output_dir (str): Output directory for the generated files
         rate (int): Rate limit, in packets/second, to apply to matched traffic, instead of a binary verdict
-- 
GitLab