From 7e4da4fd82ed036195748e82dbe22e67c03356f0 Mon Sep 17 00:00:00 2001
From: Alexey Shlyonskikh <dedantysag2@yandex.ru>
Date: Mon, 22 Apr 2024 20:54:52 +0300
Subject: [PATCH] Add missing Python types for CFD

---
 src/python_bindings/py_util/get_py_type.cpp | 2 ++
 src/python_bindings/py_util/py_to_any.cpp   | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/python_bindings/py_util/get_py_type.cpp b/src/python_bindings/py_util/get_py_type.cpp
index 3d42b53041..5255d7e59d 100644
--- a/src/python_bindings/py_util/get_py_type.cpp
+++ b/src/python_bindings/py_util/get_py_type.cpp
@@ -8,6 +8,7 @@
 #include <Python.h>
 #include <pybind11/stl/filesystem.h>
 
+#include "algorithms/cfd/enums.h"
 #include "algorithms/metric/enums.h"
 #include "association_rules/ar_algorithm_enums.h"
 #include "config/error_measure/type.h"
@@ -75,6 +76,7 @@ py::tuple GetPyType(std::type_index type_index) {
             PyTypePair<algos::metric::MetricAlgo, kPyStr>,
             PyTypePair<config::ErrorMeasureType, kPyStr>,
             PyTypePair<algos::InputFormat, kPyStr>,
+            PyTypePair<algos::cfd::Substrategy, kPyStr>,
             PyTypePair<std::vector<unsigned int>, kPyList, kPyInt>,
             {typeid(config::InputTable),
              []() { return MakeTypeTuple(py::type::of<config::InputTable>()); }},
diff --git a/src/python_bindings/py_util/py_to_any.cpp b/src/python_bindings/py_util/py_to_any.cpp
index 1e7bfa0ab9..d8eb12e923 100644
--- a/src/python_bindings/py_util/py_to_any.cpp
+++ b/src/python_bindings/py_util/py_to_any.cpp
@@ -8,6 +8,7 @@
 #include <pybind11/stl/filesystem.h>
 
 #include "algorithms/algebraic_constraints/bin_operation_enum.h"
+#include "algorithms/cfd/enums.h"
 #include "algorithms/metric/enums.h"
 #include "association_rules/ar_algorithm_enums.h"
 #include "config/error_measure/type.h"
@@ -121,6 +122,7 @@ std::unordered_map<std::type_index, ConvFunc> const kConverters{
         kEnumConvPair<algos::metric::MetricAlgo>,
         kEnumConvPair<config::ErrorMeasureType>,
         kEnumConvPair<algos::InputFormat>,
+        kEnumConvPair<algos::cfd::Substrategy>,
         kCharEnumConvPair<algos::Binop>,
         {typeid(config::InputTable), InputTableToAny},
         {typeid(config::InputTables), InputTablesToAny},