a bug fix

Author: smasuda

dicomanonymizer/simpledicomanonymizer.py

@@ -6,7 +6,16 @@
 from dataclasses import dataclass
 
 from dicomanonymizer.dicomfields import (
-    D_TAGS, Z_TAGS, X_TAGS, U_TAGS, Z_D_TAGS, X_Z_TAGS, X_D_TAGS, X_Z_D_TAGS, X_Z_U_STAR_TAGS)
+    D_TAGS,
+    Z_TAGS,
+    X_TAGS,
+    U_TAGS,
+    Z_D_TAGS,
+    X_Z_TAGS,
+    X_D_TAGS,
+    X_Z_D_TAGS,
+    X_Z_U_STAR_TAGS,
+)
 from dicomanonymizer.format_tag import tag_to_hex_strings
 
 
@@ -349,8 +358,12 @@ def initialize_actions(dicom_version: str = "2013") -> dict:
     anonymization_actions.update({tag: empty_or_replace for tag in Z_D_TAGS})
     anonymization_actions.update({tag: delete_or_empty for tag in X_Z_TAGS})
     anonymization_actions.update({tag: delete_or_replace for tag in X_D_TAGS})
-    anonymization_actions.update({tag: delete_or_empty_or_replace for tag in X_Z_D_TAGS})
-    anonymization_actions.update({tag: delete_or_empty_or_replace_UID for tag in X_Z_U_STAR_TAGS})
+    anonymization_actions.update(
+        {tag: delete_or_empty_or_replace for tag in X_Z_D_TAGS}
+    )
+    anonymization_actions.update(
+        {tag: delete_or_empty_or_replace_UID for tag in X_Z_U_STAR_TAGS}
+    )
     return anonymization_actions
 
 
@@ -443,8 +456,12 @@ def get_private_tags(
     return private_tags
 
 
-def anonymize_dataset(dataset: pydicom.Dataset, extra_anonymization_rules: dict = None,
-                      delete_private_tags: bool = True) -> None:
+def anonymize_dataset(
+    dataset: pydicom.Dataset,
+    base_rules_gen: Callable = initialize_actions,
+    extra_anonymization_rules: dict = None,
+    delete_private_tags: bool = True,
+) -> None:
     """
     Anonymize a pydicom Dataset by using anonymization rules which links an action to a tag