Skip to content

Commit 4b068e6

Browse files
committed
make get_indices private
1 parent 7d2cfbd commit 4b068e6

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/circuit.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -454,7 +454,7 @@ nb::class_<Circuit>(m, "Circuit")
454454
.def(nb::init<>())
455455
.def("add_SDD_from_file", &Circuit::add_SDD_from_file, "filename"_a, "true_lits"_a = std::vector<int>(), "false_lits"_a = std::vector<int>())
456456
.def("add_D4_from_file", &Circuit::add_D4_from_file, "filename"_a, "true_lits"_a = std::vector<int>(), "false_lits"_a = std::vector<int>())
457-
.def("get_indices", &Circuit::get_indices)
457+
.def("_get_indices", &Circuit::get_indices)
458458
.def("nb_nodes", &Circuit::nb_nodes, "number of nodes in the circuit")
459459
.def("nb_root_nodes", &Circuit::nb_root_nodes, "number of root nodes in the circuit")
460460
.def("true_node", &Circuit::true_node, "adds a true node to the circuit, and returns a pointer")

src/klay/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,13 @@
44

55
def to_torch_module(circuit: Circuit, semiring: str = "log"):
66
from .backends import torch_backend
7-
indices = circuit.get_indices()
7+
indices = circuit._get_indices()
88
return torch_backend.KnowledgeModule(*indices, semiring=semiring)
99

1010

1111
def to_jax_function(circuit: Circuit, semiring: str = "log"):
1212
from .backends import jax_backend
13-
indices = circuit.get_indices()
13+
indices = circuit._get_indices()
1414
return jax_backend.create_knowledge_layer(*indices, semiring=semiring)
1515

1616

0 commit comments

Comments
 (0)