Update optimal_transport_reweighting.py

dimitra-maoutsa · dimitra-maoutsa · commit 1ac5afcb510e · 2021-12-29T18:10:11.000+01:00
diff --git a/DeterministicParticleFlowControl/reweighting/optimal_transport_reweighting.py b/DeterministicParticleFlowControl/reweighting/optimal_transport_reweighting.py
@@ -12,6 +12,7 @@
 from pyemd import emd_with_flow
 from scipy.spatial.distance import pdist, squareform
 
+__all__ = ["reweight_optimal_transport_multidim"]
 
 def reweight_optimal_transport_multidim(samples, weights):
 
