Skip to content
Snippets Groups Projects
Commit e15c0e9a authored by Charles Villard's avatar Charles Villard
Browse files

add: pybind: add filter using bitwise enum

parent 745c5b80
No related branches found
No related tags found
No related merge requests found
...@@ -50,6 +50,7 @@ void init_GraphView(py::module& m) { ...@@ -50,6 +50,7 @@ void init_GraphView(py::module& m) {
)mydelimiter") )mydelimiter")
.def("get_input_nodes", &GraphView::inputNodes, .def("get_input_nodes", &GraphView::inputNodes,
py::arg("filter") = Aidge::InputCategory::All,
R"mydelimiter( R"mydelimiter(
Get set of input Nodes. Get set of input Nodes.
......
...@@ -28,11 +28,13 @@ void init_Operator(py::module& m){ ...@@ -28,11 +28,13 @@ void init_Operator(py::module& m){
.value("Data", OperatorType::Data) .value("Data", OperatorType::Data)
.value("Tensor", OperatorType::Tensor); .value("Tensor", OperatorType::Tensor);
py::enum_<InputCategory>(m, "InputCategory") py::enum_<Aidge::InputCategory>(m, "InputCategory", py::arithmetic())
.value("Data", InputCategory::Data) .value("Optional", Aidge::InputCategory::Optional)
.value("Param", InputCategory::Param) .value("Data", Aidge::InputCategory::Data)
.value("OptionalData", InputCategory::OptionalData) .value("OptionalData", Aidge::InputCategory::OptionalData)
.value("OptionalParam", InputCategory::OptionalParam); .value("Param", Aidge::InputCategory::Param)
.value("OptionalParam", Aidge::InputCategory::OptionalParam)
.value("All", Aidge::InputCategory::All);
py::class_<Operator, std::shared_ptr<Operator>>(m, "Operator") py::class_<Operator, std::shared_ptr<Operator>>(m, "Operator")
.def("__repr__", &Operator::repr) .def("__repr__", &Operator::repr)
......
...@@ -132,12 +132,6 @@ void init_Aidge(py::module& m) { ...@@ -132,12 +132,6 @@ void init_Aidge(py::module& m) {
init_Attributes(m); init_Attributes(m);
init_Spikegen(m); init_Spikegen(m);
init_Node(m);
init_GraphView(m);
init_OpArgs(m);
init_Connector(m);
init_SinglePassGraphMatching(m);
init_OperatorImpl(m); init_OperatorImpl(m);
init_Log(m); init_Log(m);
init_Operator(m); init_Operator(m);
...@@ -146,6 +140,12 @@ void init_Aidge(py::module& m) { ...@@ -146,6 +140,12 @@ void init_Aidge(py::module& m) {
init_StaticAnalysis(m); init_StaticAnalysis(m);
init_DynamicAnalysis(m); init_DynamicAnalysis(m);
init_Node(m);
init_GraphView(m);
init_OpArgs(m);
init_Connector(m);
init_SinglePassGraphMatching(m);
init_Abs(m); init_Abs(m);
init_Add(m); init_Add(m);
init_And(m); init_And(m);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment