Skip to content

Commit

Permalink
fixed bug in pybinds
Browse files Browse the repository at this point in the history
  • Loading branch information
SimonKlx committed Nov 7, 2024
1 parent 44cd01f commit 7274a74
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 33 deletions.
10 changes: 5 additions & 5 deletions plugins/machine_learning/python/python_bindings.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -86,25 +86,25 @@ namespace hal
.export_values();

// Bindings for NetlistGraph
py::class_<machine_learning::graph::NetlistGraph> py_netlist_graph(py_graph, "NetlistGraph", R"(
py::class_<machine_learning::NetlistGraph> py_netlist_graph(m, "NetlistGraph", R"(
Represents a graph of the netlist.
)");

py_netlist_graph.def_readwrite("edge_list", &machine_learning::graph::NetlistGraph::edge_list, R"(
py_netlist_graph.def_readwrite("edge_list", &machine_learning::NetlistGraph::edge_list, R"(
Edge list of the graph as a tuple of source and target node indices.
:type: tuple[list[int], list[int]]
)");

py_netlist_graph.def_readwrite("direction", &machine_learning::graph::NetlistGraph::direction, R"(
py_netlist_graph.def_readwrite("direction", &machine_learning::NetlistGraph::direction, R"(
Direction of the graph.
:type: hal_py.machine_learning.GraphDirection
)");

// Bindings for construct_netlist_graph
m.def("construct_netlist_graph",
&machine_learning::graph::construct_netlist_graph,
&machine_learning::construct_netlist_graph,
py::arg("netlist"),
py::arg("gates"),
py::arg("direction"),
Expand All @@ -120,7 +120,7 @@ namespace hal

// Bindings for annotate_netlist_graph
m.def("annotate_netlist_graph",
&machine_learning::graph::annotate_netlist_graph,
&machine_learning::annotate_netlist_graph,
py::arg("netlist"),
py::arg("gates"),
py::arg("netlist_graph"),
Expand Down
28 changes: 0 additions & 28 deletions plugins/machine_learning/requirements.txt

This file was deleted.

0 comments on commit 7274a74

Please sign in to comment.