expose delete_node (pytorch#12840)

Summary:
Pull Request resolved: pytorch#12840

Add binding for delete_node

Reviewed By: duc0

Differential Revision: D10453555

fbshipit-source-id: cdcaca8420a9a0c61479961d907ef6bb5478a41d

Author: BlueberryDS

caffe2/python/nomnigraph_test.py

@@ -252,6 +252,12 @@ def test_node_interactions(self):
         for node in [x, y]:
             assert node.isTensor()
 
+    def test_delete_node(self):
+        nn = ng.NNModule()
+        node = nn.dataFlow.createNode(ng.NeuralNetOperator("TestOp"))
+        nn.dataFlow.deleteNode(node)
+        assert len(nn.dataFlow.getMutableNodes()) == 0
+
     def test_annotation_basic(self):
         annot = ng.Annotation()
         annot.setDevice("woot")

caffe2/python/pybind_state_nomni.cc

@@ -216,6 +216,7 @@ void addNomnigraphMethods(pybind11::module& m) {
             return g->createNode(convertToNeuralNetOperator(op));
           },
           py::return_value_policy::reference_internal)
+      .def("deleteNode", &NNGraph::deleteNode)
       .def(
           "getMutableNodes",
           &NNGraph::getMutableNodes,