Pre commit before rebase magic

tomsch420 · Oct 11, 2024 · b1963c8 · b1963c8
1 parent bd7f3e3
commit b1963c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/probabilistic_model/probabilistic_circuit/jax/inner_layer.py b/src/probabilistic_model/probabilistic_circuit/jax/inner_layer.py
@@ -432,7 +432,7 @@ def sample_from_frequencies(self, frequencies: np.array, result: np.array, start
             current_frequency_block = node_to_child_frequency_map[:,
                                       prev_column_index:prev_column_index + child_layer.number_of_nodes]
             frequencies_for_child_nodes = current_frequency_block.sum(0)
-            child_layer.sample_from_frequencies2(frequencies_for_child_nodes, result, consumed_indices)
+            child_layer.sample_from_frequencies(frequencies_for_child_nodes, result, consumed_indices)
             consumed_indices += frequencies_for_child_nodes.sum()
 
             # shift the offset