From 7274a7427d3216893f5deb27791c5347336fd6ea Mon Sep 17 00:00:00 2001 From: Simon Klix Date: Thu, 7 Nov 2024 18:52:27 +0100 Subject: [PATCH] fixed bug in pybinds --- .../python/python_bindings.cpp | 10 +++---- plugins/machine_learning/requirements.txt | 28 ------------------- 2 files changed, 5 insertions(+), 33 deletions(-) delete mode 100644 plugins/machine_learning/requirements.txt diff --git a/plugins/machine_learning/python/python_bindings.cpp b/plugins/machine_learning/python/python_bindings.cpp index e63449a4f3c..e10ff8be015 100644 --- a/plugins/machine_learning/python/python_bindings.cpp +++ b/plugins/machine_learning/python/python_bindings.cpp @@ -86,17 +86,17 @@ namespace hal .export_values(); // Bindings for NetlistGraph - py::class_ py_netlist_graph(py_graph, "NetlistGraph", R"( + py::class_ 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 @@ -104,7 +104,7 @@ namespace hal // 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"), @@ -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"), diff --git a/plugins/machine_learning/requirements.txt b/plugins/machine_learning/requirements.txt deleted file mode 100644 index bb0aa0b6713..00000000000 --- a/plugins/machine_learning/requirements.txt +++ /dev/null @@ -1,28 +0,0 @@ -filelock==3.16.1 -fsspec==2024.10.0 -Jinja2==3.1.4 -MarkupSafe==3.0.2 -mpmath==1.3.0 -networkx==3.4.2 -numpy==2.1.3 -nvidia-cublas-cu12==12.4.5.8 -nvidia-cuda-cupti-cu12==12.4.127 -nvidia-cuda-nvrtc-cu12==12.4.127 -nvidia-cuda-runtime-cu12==12.4.127 -nvidia-cudnn-cu12==9.1.0.70 -nvidia-cufft-cu12==11.2.1.3 -nvidia-curand-cu12==10.3.5.147 -nvidia-cusolver-cu12==11.6.1.9 -nvidia-cusparse-cu12==12.3.1.170 -nvidia-nccl-cu12==2.21.5 -nvidia-nvjitlink-cu12==12.4.127 -nvidia-nvtx-cu12==12.4.127 -pillow==11.0.0 -python-dotenv==1.0.1 -setuptools==75.3.0 -sympy==1.13.1 -torch==2.5.1 -torchaudio==2.5.1 -torchvision==0.20.1 -triton==3.1.0 -typing_extensions==4.12.2