fixed missing val_from_arch parameter and pybrush breaking

cavalab · May 4, 2024 · 78314fc · 78314fc
1 parent 12a8772
commit 78314fc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/bindings/bind_engines.h b/src/bindings/bind_engines.h
@@ -18,6 +18,9 @@
 #include "../pop/population.cpp"
 #include "../pop/population.h"
 
+#include "../pop/archive.cpp"
+#include "../pop/archive.h"
+
 using Reg = Brush::RegressorEngine;
 using Cls = Brush::ClassifierEngine;
 using Rep = Brush::RepresenterEngine;

diff --git a/src/bindings/bind_params.cpp b/src/bindings/bind_params.cpp
@@ -26,6 +26,7 @@ void bind_params(py::module& m)
         .def_property("logfile", &Brush::Parameters::get_logfile, &Brush::Parameters::set_logfile)
         .def_property("num_islands", &Brush::Parameters::get_num_islands, &Brush::Parameters::set_num_islands)
         .def_property("use_arch", &Brush::Parameters::get_use_arch, &Brush::Parameters::set_use_arch)
+        .def_property("val_from_arch", &Brush::Parameters::get_val_from_arch, &Brush::Parameters::set_val_from_arch)
         .def_property("n_classes", &Brush::Parameters::get_n_classes, &Brush::Parameters::set_n_classes)
         .def_property("n_jobs", &Brush::Parameters::get_n_jobs, &Brush::Parameters::set_n_classes)
         .def_property("classification", &Brush::Parameters::get_classification, &Brush::Parameters::set_classification)