eryn 1.2.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- eryn/CMakeLists.txt +51 -0
- eryn/__init__.py +35 -0
- eryn/backends/__init__.py +20 -0
- eryn/backends/backend.py +1150 -0
- eryn/backends/hdfbackend.py +819 -0
- eryn/ensemble.py +1690 -0
- eryn/git_version.py.in +7 -0
- eryn/model.py +18 -0
- eryn/moves/__init__.py +42 -0
- eryn/moves/combine.py +135 -0
- eryn/moves/delayedrejection.py +229 -0
- eryn/moves/distgen.py +104 -0
- eryn/moves/distgenrj.py +222 -0
- eryn/moves/gaussian.py +190 -0
- eryn/moves/group.py +281 -0
- eryn/moves/groupstretch.py +120 -0
- eryn/moves/mh.py +193 -0
- eryn/moves/move.py +703 -0
- eryn/moves/mtdistgen.py +137 -0
- eryn/moves/mtdistgenrj.py +190 -0
- eryn/moves/multipletry.py +776 -0
- eryn/moves/red_blue.py +333 -0
- eryn/moves/rj.py +388 -0
- eryn/moves/stretch.py +231 -0
- eryn/moves/tempering.py +649 -0
- eryn/pbar.py +56 -0
- eryn/prior.py +452 -0
- eryn/state.py +775 -0
- eryn/tests/__init__.py +0 -0
- eryn/tests/test_eryn.py +1246 -0
- eryn/utils/__init__.py +10 -0
- eryn/utils/periodic.py +134 -0
- eryn/utils/stopping.py +164 -0
- eryn/utils/transform.py +226 -0
- eryn/utils/updates.py +69 -0
- eryn/utils/utility.py +329 -0
- eryn-1.2.0.dist-info/METADATA +167 -0
- eryn-1.2.0.dist-info/RECORD +39 -0
- eryn-1.2.0.dist-info/WHEEL +4 -0
eryn/CMakeLists.txt
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
# Add subdirectories which contain compiled sources
|
|
2
|
+
|
|
3
|
+
# Handle Git Metadata
|
|
4
|
+
get_target_property(_ERYN_WITH_CPU eryn WITH_CPU)
|
|
5
|
+
get_target_property(_ERYN_WITH_GPU eryn WITH_GPU)
|
|
6
|
+
|
|
7
|
+
message(CHECK_START "Building eryn.git_version metadata")
|
|
8
|
+
message(CHECK_START "Find Git executable")
|
|
9
|
+
find_package(Git)
|
|
10
|
+
if(Git_FOUND)
|
|
11
|
+
message(CHECK_PASS "found")
|
|
12
|
+
message(CHECK_START "Checking if in Git worktree")
|
|
13
|
+
execute_process(
|
|
14
|
+
COMMAND ${GIT_EXECUTABLE} rev-parse --is-inside-work-tree
|
|
15
|
+
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
|
|
16
|
+
OUTPUT_QUIET ERROR_QUIET
|
|
17
|
+
RESULT_VARIABLE ERYN_GIT_IN_WORKTREE_CODE)
|
|
18
|
+
if(ERYN_GIT_IN_WORKTREE_CODE EQUAL "0")
|
|
19
|
+
message(CHECK_PASS "yes")
|
|
20
|
+
|
|
21
|
+
execute_process(
|
|
22
|
+
COMMAND ${GIT_EXECUTABLE} log -1 --format=%H
|
|
23
|
+
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
|
|
24
|
+
OUTPUT_VARIABLE ERYN_GIT_COMMIT_ID
|
|
25
|
+
OUTPUT_STRIP_TRAILING_WHITESPACE)
|
|
26
|
+
execute_process(
|
|
27
|
+
COMMAND ${GIT_EXECUTABLE} log -1 --format=%h
|
|
28
|
+
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
|
|
29
|
+
OUTPUT_VARIABLE ERYN_GIT_COMMIT_SHORT_ID
|
|
30
|
+
OUTPUT_STRIP_TRAILING_WHITESPACE)
|
|
31
|
+
|
|
32
|
+
configure_file(git_version.py.in git_version.py @ONLY)
|
|
33
|
+
|
|
34
|
+
if(_ERYN_WITH_CPU)
|
|
35
|
+
install(FILES ${CMAKE_CURRENT_BINARY_DIR}/git_version.py DESTINATION eryn)
|
|
36
|
+
install(FILES ${CMAKE_CURRENT_BINARY_DIR}/git_version.py
|
|
37
|
+
DESTINATION eryn_backend_cpu)
|
|
38
|
+
endif()
|
|
39
|
+
if(_ERYN_WITH_GPU)
|
|
40
|
+
install(FILES ${CMAKE_CURRENT_BINARY_DIR}/git_version.py
|
|
41
|
+
DESTINATION eryn_backend_cuda${CUDAToolkit_VERSION_MAJOR}x)
|
|
42
|
+
endif()
|
|
43
|
+
|
|
44
|
+
message(CHECK_PASS "done")
|
|
45
|
+
else()
|
|
46
|
+
message(CHECK_FAIL "no")
|
|
47
|
+
endif()
|
|
48
|
+
else()
|
|
49
|
+
message(CHECK_FAIL "not found")
|
|
50
|
+
message(CHECK_FAIL "eryn.git_version will not be available")
|
|
51
|
+
endif()
|
eryn/__init__.py
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
"""GPU Backend Tools."""
|
|
2
|
+
|
|
3
|
+
# ruff: noqa: E402
|
|
4
|
+
try:
|
|
5
|
+
from eryn._version import ( # pylint: disable=E0401,E0611
|
|
6
|
+
__version__,
|
|
7
|
+
__version_tuple__,
|
|
8
|
+
)
|
|
9
|
+
|
|
10
|
+
except ModuleNotFoundError:
|
|
11
|
+
from importlib.metadata import PackageNotFoundError, version # pragma: no cover
|
|
12
|
+
|
|
13
|
+
try:
|
|
14
|
+
__version__ = version(__name__)
|
|
15
|
+
__version_tuple__ = tuple(__version__.split("."))
|
|
16
|
+
except PackageNotFoundError: # pragma: no cover
|
|
17
|
+
__version__ = "unknown"
|
|
18
|
+
__version_tuple__ = (0, 0, 0, "unknown")
|
|
19
|
+
finally:
|
|
20
|
+
del version, PackageNotFoundError
|
|
21
|
+
|
|
22
|
+
_is_editable: bool
|
|
23
|
+
try:
|
|
24
|
+
from . import _editable
|
|
25
|
+
|
|
26
|
+
_is_editable = True
|
|
27
|
+
del _editable
|
|
28
|
+
except (ModuleNotFoundError, ImportError):
|
|
29
|
+
_is_editable = False
|
|
30
|
+
|
|
31
|
+
__all__ = [
|
|
32
|
+
"__version__",
|
|
33
|
+
"__version_tuple__",
|
|
34
|
+
"_is_editable",
|
|
35
|
+
]
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
|
|
3
|
+
from .backend import Backend
|
|
4
|
+
|
|
5
|
+
from .hdfbackend import HDFBackend, TempHDFBackend
|
|
6
|
+
|
|
7
|
+
__all__ = ["Backend", "HDFBackend", "TempHDFBackend", "get_test_backends"]
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
def get_test_backends():
|
|
11
|
+
backends = [Backend]
|
|
12
|
+
|
|
13
|
+
try:
|
|
14
|
+
import h5py # NOQA
|
|
15
|
+
except ImportError:
|
|
16
|
+
pass
|
|
17
|
+
else:
|
|
18
|
+
backends.append(TempHDFBackend)
|
|
19
|
+
|
|
20
|
+
return backends
|