File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -133,7 +133,6 @@ def sample(
133133 for score_key in samples .keys ():
134134 samples [score_key ] = self .standardize_layers ["inference_variables" ](samples [score_key ], forward = False )
135135
136- samples = {"inference_variables" : samples }
137136 samples = self ._apply_inverse_adapter_to_samples (samples , ** kwargs )
138137
139138 if split :
@@ -218,9 +217,9 @@ def _apply_inverse_adapter_to_samples(
218217 """Applies the inverse adapter to a dictionary of samples."""
219218 samples = keras .tree .map_structure (keras .ops .convert_to_numpy , samples )
220219 processed = {}
221- for score_key , samples in samples .items ():
220+ for score_key , score_value in samples .items ():
222221 processed [score_key ] = self .adapter (
223- {"inference_variables" : samples },
222+ {"inference_variables" : score_value },
224223 inverse = True ,
225224 strict = False ,
226225 ** kwargs ,
You can’t perform that action at this time.
0 commit comments