PKG: Suppress error when two version of OpenMP are loaded

Author: RUrlus

src/diptest/__init__.py

@@ -1,11 +1,30 @@
 from __future__ import annotations
 
 import importlib.metadata
+import os
 
 from diptest.diptest import dipstat, diptest
 from diptest.lib import _diptest_core as _diptest
 from diptest.lib._diptest_core import _has_openmp_support
 
+# Setting the following environment variable allows multiple OpenMP
+# libraries to be loaded. This is also used without issue by Scikit-learn.
+# OpenMP error msg:
+# /* OMP: Error #15: Initializing libomp.dylib, but found libomp.dylib already initialized.
+#  * OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program.
+#  * That is dangerous, since it can degrade performance or cause incorrect results.
+#  * The best thing to do is to ensure that only a single OpenMP runtime is linked into the process,
+#  * e.g. by avoiding static linking of the OpenMP runtime in any library.
+#  * As an unsafe, unsupported, undocumented workaround you can set the environment variable KMP_DUPLICATE_LIB_OK=TRUE
+#  * to allow the program to continue to execute, but that may cause crashes or silently produce incorrect results.
+#  * For more information, please see http://openmp.llvm.org/
+#  */
+os.environ.setdefault("KMP_DUPLICATE_LIB_OK", "True")
+
+# Workaround issue discovered in intel-openmp 2019.5:
+# https://github.com/ContinuumIO/anaconda-issues/issues/11294
+os.environ.setdefault("KMP_INIT_AT_FORK", "FALSE")
+
 __version__ = importlib.metadata.version("diptest")
 
 __all__ = ["dipstat", "diptest", "_diptest", "_has_openmp_support", "__version__"]