bindings

kennethassogba · Nov 24, 2024 · 2727280 · 2727280
1 parent 1f8d203
commit 2727280
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 18 deletions.
diff --git a/src/demeter/model/material.bind.cpp b/src/demeter/model/material.bind.cpp
@@ -16,25 +16,20 @@ using Eigen::ArrayXXd;
 NB_MODULE(_demeter, m) {
   m.doc() = "A deterministic neutron transport solver for reactor simulations";
 
-  // Material class bindings
   // clang-format off
+  // Material class bindings
   nb::class_<Demeter::Material>(m, "Material")
-      .def(nb::init<ArrayXd&, /* sigma_t */
+      .def(nb::init<ArrayXd&,  /* sigma_t */
                     ArrayXXd&, /* sigma_s */
-                    ArrayXd&, /* sigma_a */
-                    ArrayXd&, /* sigma_f */
-                    ArrayXd&, /* nu_sigma_f */
-                    ArrayXd&, /* chi */
+                    ArrayXd&,  /* sigma_a */
+                    ArrayXd&,  /* sigma_f */
+                    ArrayXd&,  /* nu_sigma_f */
+                    ArrayXd&,  /* chi */
                     std::string_view>(),
-           "sigma_t"_a, "sigma_s"_a, "sigma_a"_a, "sigma_f"_a, "nu_sigma_f"_a,
-           "chi"_a, "name"_a = "",
-           "Create a Material object with the given "
-           "cross-sections.");
-  // doublon
-  // .def(nb::init<ArrayXd&&, ArrayXXd&&, ArrayXd&&, ArrayXd&&, ArrayXd&&,
-  //               ArrayXd&&, std::string_view>(),
-  //      "sigma_t"_a, "sigma_s"_a, "sigma_a"_a, "sigma_f"_a, "nu_sigma_f"_a,
-  //      "chi"_a, "name"_a = "");
+           "sigma_t"_a, "sigma_s"_a, "sigma_a"_a,
+           "sigma_f"_a, "nu_sigma_f"_a, "chi"_a,
+           "name"_a = "",
+           "Create a Material object with the given cross-sections.");
   // .def_ro("name", &Demeter::Material::name, "The name of the material.")
   // .def_ro("num_energy_groups", &Demeter::Material::NumEnergyGroups);
   // .def_prop_ro("sigma_t", &Demeter::Material::SigmaT)

diff --git a/tests/regress/test00.py b/tests/regress/test00.py
@@ -2,8 +2,6 @@
 import numpy as np
 import matplotlib.pyplot as plt
 
-# "sigma_t"_a, "sigma_s"_a, "sigma_a"_a, "sigma_f"_a, "nu_sigma_f"_a,
-          #  "chi"_a, "name"_a
 # Create materials
 sigma_t = np.array([0.222222, 0.833333])
 sigma_s = np.array([[0.00, 0.02],
@@ -13,7 +11,7 @@
 nu_sigma_f = np.array([0., 0.135])
 chi = np.array([1., 0.])
 
-uO2 = Material(sigma_t, sigma_s, sigma_a, sigma_f, nu_sigma_f, chi, name= 'uO2')
+uO2 = Material(sigma_t, sigma_s, sigma_a, sigma_f, nu_sigma_f, chi, name= 'uO2') # type: ignore
 
 """
 sigma_t = np.array([0.1, 0.2, 0.3])