@@ -17,7 +17,6 @@ def evaluate_pipeline(nodes: Sequence[ConversionNode], input: dict[str, Any]):
1717
1818@dataclass
1919class ConversionNode :
20- name : str
2120 required_keys : tuple [str , ...]
2221 output_keys : tuple [str , ...]
2322 trim_keys : bool
@@ -43,12 +42,12 @@ class UnionConversionNode(ConversionNode):
4342 nodes : tuple [ConversionNode , ...]
4443
4544 @classmethod
46- def from_nodes (cls , name : str , * nodes : ConversionNode , trim_keys = False ):
45+ def from_nodes (cls , * nodes : ConversionNode , trim_keys = False ):
4746 required = tuple (set (k for n in nodes for k in n .required_keys ))
4847 output = Counter (k for n in nodes for k in n .output_keys )
4948 if duplicate := {k for k , v in output .items () if v > 1 }:
5049 raise ValueError (f"Duplicate keys from multiple input nodes: { duplicate } " )
51- return cls (name , required , tuple (output ), trim_keys , nodes )
50+ return cls (required , tuple (output ), trim_keys , nodes )
5251
5352 def evaluate (self , input : dict [str , Any ]) -> dict [str , Any ]:
5453 return super ().evaluate ({k : v for n in self .nodes for k , v in n .evaluate (input ).items ()})
@@ -59,12 +58,12 @@ class RenameConversionNode(ConversionNode):
5958 mapping : dict [str , str ]
6059
6160 @classmethod
62- def from_mapping (cls , name : str , mapping : dict [str , str ], trim_keys = False ):
61+ def from_mapping (cls , mapping : dict [str , str ], trim_keys = False ):
6362 required = tuple (mapping )
6463 output = Counter (mapping .values ())
6564 if duplicate := {k for k , v in output .items () if v > 1 }:
6665 raise ValueError (f"Duplicate output keys in mapping: { duplicate } " )
67- return cls (name , required , tuple (output ), trim_keys , mapping )
66+ return cls (required , tuple (output ), trim_keys , mapping )
6867
6968 def evaluate (self , input : dict [str , Any ]) -> dict [str , Any ]:
7069 return super ().evaluate ({** input , ** {out : input [inp ] for (inp , out ) in self .mapping .items ()}})
@@ -79,14 +78,14 @@ def _sigs(self):
7978 return {k : (f , inspect .signature (f )) for k , f in self .funcs .items ()}
8079
8180 @classmethod
82- def from_funcs (cls , name : str , funcs : dict [str , Callable ], trim_keys = False ):
81+ def from_funcs (cls , funcs : dict [str , Callable ], trim_keys = False ):
8382 sigs = {k : inspect .signature (f ) for k , f in funcs .items ()}
8483 output = tuple (sigs )
8584 input = []
8685 for v in sigs .values ():
8786 input .extend (v .parameters .keys ())
8887 input = tuple (set (input ))
89- return cls (name , input , output , trim_keys , funcs )
88+ return cls (input , output , trim_keys , funcs )
9089
9190 def evaluate (self , input : dict [str , Any ]) -> dict [str , Any ]:
9291 return super ().evaluate (
@@ -102,8 +101,8 @@ class LimitKeysConversionNode(ConversionNode):
102101 keys : set [str ]
103102
104103 @classmethod
105- def from_keys (cls , name : str , keys : Sequence [str ]):
106- return cls (name , (), tuple (keys ), trim_keys = True , keys = set (keys ))
104+ def from_keys (cls , keys : Sequence [str ]):
105+ return cls ((), tuple (keys ), trim_keys = True , keys = set (keys ))
107106
108107 def evaluate (self , input : dict [str , Any ]) -> dict [str , Any ]:
109108 return {k : v for k , v in input .items () if k in self .keys }
0 commit comments