summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMarc Glisse <marc.glisse@inria.fr>2020-02-11 13:06:48 +0100
committerMarc Glisse <marc.glisse@inria.fr>2020-02-11 13:06:48 +0100
commitd6f3165831d20bf3a91f1ff7e9734a574eaa567a (patch)
tree982a27da9994fa7f0b93921a02f4a4fd04093255
parentb75123eeda446e7f778d4939da67a78e4c8c6abc (diff)
License and author
-rw-r--r--src/python/gudhi/hera.cc13
-rwxr-xr-xsrc/python/test/test_wasserstein_distance.py2
2 files changed, 12 insertions, 3 deletions
diff --git a/src/python/gudhi/hera.cc b/src/python/gudhi/hera.cc
index 61f0da10..0d562b4c 100644
--- a/src/python/gudhi/hera.cc
+++ b/src/python/gudhi/hera.cc
@@ -1,9 +1,19 @@
+/* This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT.
+ * See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details.
+ * Author(s): Marc Glisse
+ *
+ * Copyright (C) 2020 Inria
+ *
+ * Modification(s):
+ * - YYYY/MM Author: Description of the modification
+ */
+
#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
#include <boost/range/iterator_range.hpp>
-#include <wasserstein.h>
+#include <wasserstein.h> // Hera
#include <array>
@@ -41,7 +51,6 @@ double wasserstein_distance(
PYBIND11_MODULE(hera, m) {
m.def("wasserstein_distance", &wasserstein_distance,
py::arg("X"), py::arg("Y"),
- // Should we name those q, p and d instead?
py::arg("order") = 1,
py::arg("internal_p") = std::numeric_limits<double>::infinity(),
py::arg("delta") = .01,
diff --git a/src/python/test/test_wasserstein_distance.py b/src/python/test/test_wasserstein_distance.py
index 4bc7114e..6a6b217b 100755
--- a/src/python/test/test_wasserstein_distance.py
+++ b/src/python/test/test_wasserstein_distance.py
@@ -1,6 +1,6 @@
""" This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT.
See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details.
- Author(s): Theo Lacombe
+ Author(s): Theo Lacombe, Marc Glisse
Copyright (C) 2019 Inria