remove test code

Qihoo360 · Oct 24, 2024 · 3f4a2ee · 3f4a2ee
1 parent 7fd2871
commit 3f4a2ee
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/core/main/py_wrapper.cc b/core/main/py_wrapper.cc
@@ -114,14 +114,10 @@ PYBIND11_MODULE(_pywrap_tn, m) {
 
         return py::reinterpret_steal<py::object>(obj);
     })
-    .def("create_sparse_table", [](py::object obj, std::string name, int dimension, bool use_cvm) {
+    .def("create_sparse_table", [](py::object obj, std::string name, int dimension) {
         OptimizerBase* opt =
                static_cast<OptimizerBase*>(PyCapsule_GetPointer(obj.ptr(), nullptr));
 
-        opt->SetUseCvm(use_cvm);
-
-        std::cout << "Cvm plugin is: " << opt->ShouldUseCvm() << std::endl;
-
         PsCluster* cluster = PsCluster::Instance();
 
         SparseTable* table = CreateSparseTable(opt, name, dimension, cluster->RankNum(), cluster->Rank());