Merge pull request #370 from MeasureTransport/dannys4/julia-triangula…

…r-movecoeffs Add move coeffs positional argument for julia binding, fix CI
MeasureTransport · Jan 22, 2024 · 1b1e5df · 1b1e5df
2 parents fd3ebeb + 52bcfc6
commit 1b1e5df
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/bindings/julia/src/TriangularMap.cpp b/bindings/julia/src/TriangularMap.cpp
@@ -17,8 +17,8 @@ void mpart::binding::TriangularMapWrapper(jlcxx::Module &mod) {
        .method("GetComponent", &TriangularMap<Kokkos::HostSpace>::GetComponent)
     ;
 
-    mod.method("TriangularMap", [](std::vector<std::shared_ptr<ConditionalMapBase<Kokkos::HostSpace>>> vec){
-        return std::static_pointer_cast<ConditionalMapBase<Kokkos::HostSpace>>(std::make_shared<TriangularMap<Kokkos::HostSpace>>(vec));
+    mod.method("TriangularMap", [](std::vector<std::shared_ptr<ConditionalMapBase<Kokkos::HostSpace>>> vec, bool move_coeffs=false){
+        return std::static_pointer_cast<ConditionalMapBase<Kokkos::HostSpace>>(std::make_shared<TriangularMap<Kokkos::HostSpace>>(vec, move_coeffs));
     });
 
 }
diff --git a/tests/Distributions/Test_Distributions_Common.h b/tests/Distributions/Test_Distributions_Common.h
@@ -2,6 +2,7 @@
 #define TEST_DISTRIBUTIONS_COMMON_H
 
 #include<algorithm>
+#include<numeric>
 #include <catch2/catch_all.hpp>
 #include "MParT/Utilities/ArrayConversions.h"
 #include "MParT/Distributions/DensityBase.h"