Adding more flexibility to the MultiFoldTagger python config

FTagDumper/python/blocks/MultifoldTagger.py

@@ -15,6 +15,9 @@ class MultifoldTagger(BaseBlock):
     ----------
     nn_paths : list[str]
         List of paths to the neural network files.
+    alg_name : str
+        Name of the algorithm. If None, the name will be
+        extracted from the nn_paths. By default None
     target : str
         Whether to tag the BTagging object or the Jet.
     jet_collection : str | None
@@ -26,13 +29,17 @@ class MultifoldTagger(BaseBlock):
         Name of the fold hash variable.
     constituents : str
         Name of the constituent container.
+    default_zero_tracks : bool
+        Bool to decide how to work with zero tracks. By default False.
     """
     nn_paths: list[str]
+    alg_name: str = None
     target: str = "BTagging"
     jet_collection: str = None
     remap: dict = None
     fold_hash_name: str = "jetFoldHash"
     constituents: str = "InDetTrackParticles"
+    default_zero_tracks: bool = False
 
     def __post_init__(self):
 
@@ -52,25 +59,27 @@ class MultifoldTagger(BaseBlock):
             self.track_link_type = "IPARTICLE"
         else:
             raise ValueError(f"Unknown target {self.target}")
-
-        self.name = '_'.join(Path(self.nn_paths[0]).parts[2:-1])
+        
+        if self.alg_name is None:
+            self.alg_name = '_'.join(Path(self.nn_paths[0]).parts[2:-1])
 
         if "BTagTrackToJetAssociator" in self.remap:
-            self.name = f'{self.remap["BTagTrackToJetAssociator"]}_{self.name}' 
+            self.alg_name = f'{self.remap["BTagTrackToJetAssociator"]}_{self.alg_name}' 
   
     def to_ca(self):
         ca = ComponentAccumulator()
         ca.addEventAlgo(
             self.deco_alg(
-                name=f'{self.name}_Alg',
+                name=f'{self.alg_name}_Alg',
                 container=self.container,
                 constituentContainer=self.constituents,
                 decorator=CompFactory.FlavorTagDiscriminants.MultifoldGNNTool(
-                    name=f'{self.name}_Tool',
+                    name=f'{self.alg_name}_Tool',
                     foldHashName=self.fold_hash_name,
                     nnFiles=self.nn_paths,
                     variableRemapping=self.remap,
-                    trackLinkType=self.track_link_type
+                    trackLinkType=self.track_link_type,
+                    defaultZeroTracks=self.default_zero_tracks,
                 )
             )
         )