MEDS-extract 0.1__tar.gz

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.
Files changed (58) hide show
  1. meds_extract-0.1/.editorconfig +13 -0
  2. meds_extract-0.1/.github/workflows/code-quality-main.yaml +31 -0
  3. meds_extract-0.1/.github/workflows/code-quality-pr.yaml +45 -0
  4. meds_extract-0.1/.github/workflows/python-build.yaml +99 -0
  5. meds_extract-0.1/.github/workflows/tests.yaml +48 -0
  6. meds_extract-0.1/.gitignore +160 -0
  7. meds_extract-0.1/.pre-commit-config.yaml +133 -0
  8. meds_extract-0.1/.readthedocs.yaml +21 -0
  9. meds_extract-0.1/LICENSE +21 -0
  10. meds_extract-0.1/PKG-INFO +315 -0
  11. meds_extract-0.1/README.md +273 -0
  12. meds_extract-0.1/conftest.py +21 -0
  13. meds_extract-0.1/docs/gen_ref_pages.py +43 -0
  14. meds_extract-0.1/docs/index.md +1 -0
  15. meds_extract-0.1/docs/javascripts/mathjax.js +19 -0
  16. meds_extract-0.1/docs/pipeline_configuration.md +82 -0
  17. meds_extract-0.1/docs/preprocessing_operation_prototypes.md +264 -0
  18. meds_extract-0.1/docs/terminology.md +48 -0
  19. meds_extract-0.1/docs/tokenization_tensorization.md +276 -0
  20. meds_extract-0.1/mkdocs.yml +46 -0
  21. meds_extract-0.1/pyproject.toml +56 -0
  22. meds_extract-0.1/setup.cfg +4 -0
  23. meds_extract-0.1/src/MEDS_extract/__init__.py +50 -0
  24. meds_extract-0.1/src/MEDS_extract/configs/__init__.py +0 -0
  25. meds_extract-0.1/src/MEDS_extract/configs/_extract.yaml +49 -0
  26. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/__init__.py +0 -0
  27. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/convert_to_sharded_events.yaml +2 -0
  28. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/extract_code_metadata.yaml +3 -0
  29. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/finalize_MEDS_data.yaml +2 -0
  30. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/finalize_MEDS_metadata.yaml +3 -0
  31. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/merge_to_MEDS_cohort.yaml +3 -0
  32. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/shard_events.yaml +3 -0
  33. meds_extract-0.1/src/MEDS_extract/configs/stage_configs/split_and_shard_subjects.yaml +9 -0
  34. meds_extract-0.1/src/MEDS_extract/convert_to_sharded_events.py +907 -0
  35. meds_extract-0.1/src/MEDS_extract/extract_code_metadata.py +456 -0
  36. meds_extract-0.1/src/MEDS_extract/finalize_MEDS_data.py +138 -0
  37. meds_extract-0.1/src/MEDS_extract/finalize_MEDS_metadata.py +229 -0
  38. meds_extract-0.1/src/MEDS_extract/merge_to_MEDS_cohort.py +283 -0
  39. meds_extract-0.1/src/MEDS_extract/shard_events.py +426 -0
  40. meds_extract-0.1/src/MEDS_extract/split_and_shard_subjects.py +296 -0
  41. meds_extract-0.1/src/MEDS_extract/utils.py +133 -0
  42. meds_extract-0.1/src/MEDS_extract.egg-info/PKG-INFO +315 -0
  43. meds_extract-0.1/src/MEDS_extract.egg-info/SOURCES.txt +56 -0
  44. meds_extract-0.1/src/MEDS_extract.egg-info/dependency_links.txt +1 -0
  45. meds_extract-0.1/src/MEDS_extract.egg-info/entry_points.txt +8 -0
  46. meds_extract-0.1/src/MEDS_extract.egg-info/requires.txt +32 -0
  47. meds_extract-0.1/src/MEDS_extract.egg-info/top_level.txt +1 -0
  48. meds_extract-0.1/tests/__init__.py +24 -0
  49. meds_extract-0.1/tests/test_convert_to_sharded_events.py +374 -0
  50. meds_extract-0.1/tests/test_extract.py +651 -0
  51. meds_extract-0.1/tests/test_extract_code_metadata.py +220 -0
  52. meds_extract-0.1/tests/test_extract_no_metadata.py +624 -0
  53. meds_extract-0.1/tests/test_finalize_MEDS_data.py +111 -0
  54. meds_extract-0.1/tests/test_finalize_MEDS_metadata.py +107 -0
  55. meds_extract-0.1/tests/test_merge_to_MEDS_cohort.py +283 -0
  56. meds_extract-0.1/tests/test_shard_events.py +174 -0
  57. meds_extract-0.1/tests/test_split_and_shard_subjects.py +227 -0
  58. meds_extract-0.1/tests/utils.py +484 -0
@@ -0,0 +1,13 @@
1
+ root = true
2
+
3
+ [*]
4
+ charset = utf-8
5
+ end_of_line = lf
6
+ indent_size = 4
7
+ indent_style = space
8
+ insert_final_newline = true
9
+ max_line_length = 110
10
+ tab_width = 4
11
+
12
+ [{*.yaml,*.yml}]
13
+ indent_size = 2
@@ -0,0 +1,31 @@
1
+ # Same as `code-quality-pr.yaml` but triggered on commit to main branch
2
+ # and runs on all files (instead of only the changed ones)
3
+
4
+ name: Code Quality Main
5
+
6
+ on:
7
+ push:
8
+ branches: [main]
9
+
10
+ jobs:
11
+ code-quality:
12
+ runs-on: ubuntu-latest
13
+
14
+ strategy:
15
+ matrix:
16
+ python-version: ["3.12"]
17
+
18
+ steps:
19
+ - uses: actions/checkout@v4
20
+
21
+ - name: Set up Python ${{ matrix.python-version }}
22
+ uses: actions/setup-python@v5
23
+ with:
24
+ python-version: ${{ matrix.python-version }}
25
+
26
+ - name: Install packages
27
+ run: |
28
+ pip install .[dev]
29
+
30
+ - name: Run pre-commits
31
+ uses: pre-commit/action@v3.0.1
@@ -0,0 +1,45 @@
1
+ # This workflow finds which files were changed, prints them,
2
+ # and runs `pre-commit` on those files.
3
+
4
+ # Inspired by the sktime library:
5
+ # https://github.com/alan-turing-institute/sktime/blob/main/.github/workflows/test.yml
6
+
7
+ name: Code Quality PR
8
+
9
+ on:
10
+ pull_request:
11
+ branches: [main, "release/*", "dev"]
12
+
13
+ jobs:
14
+ code-quality:
15
+ runs-on: ubuntu-latest
16
+
17
+ strategy:
18
+ matrix:
19
+ python-version: ["3.12"]
20
+
21
+ steps:
22
+ - uses: actions/checkout@v4
23
+
24
+ - name: Set up Python ${{ matrix.python-version }}
25
+ uses: actions/setup-python@v5
26
+ with:
27
+ python-version: ${{ matrix.python-version }}
28
+
29
+ - name: Install packages
30
+ run: |
31
+ pip install .[dev]
32
+
33
+ - name: Find modified files
34
+ id: file_changes
35
+ uses: trilom/file-changes-action@v1.2.4
36
+ with:
37
+ output: " "
38
+
39
+ - name: List modified files
40
+ run: echo '${{ steps.file_changes.outputs.files}}'
41
+
42
+ - name: Run pre-commits
43
+ uses: pre-commit/action@v3.0.1
44
+ with:
45
+ extra_args: --files ${{ steps.file_changes.outputs.files}}
@@ -0,0 +1,99 @@
1
+ name: Publish Python 🐍 distribution 📦 to PyPI and TestPyPI
2
+
3
+ on: push
4
+
5
+ jobs:
6
+ build:
7
+ name: Build distribution 📦
8
+ runs-on: ubuntu-latest
9
+
10
+ strategy:
11
+ matrix:
12
+ python-version: ["3.12"]
13
+
14
+ steps:
15
+ - uses: actions/checkout@v4
16
+ - name: Set up Python ${{ matrix.python-version }}
17
+ uses: actions/setup-python@v5
18
+ with:
19
+ python-version: ${{ matrix.python-version }}
20
+ - name: Install pypa/build
21
+ run: >-
22
+ python3 -m
23
+ pip install
24
+ build
25
+ --user
26
+ - name: Build a binary wheel and a source tarball
27
+ run: python3 -m build
28
+ - name: Store the distribution packages
29
+ uses: actions/upload-artifact@v4
30
+ with:
31
+ name: python-package-distributions
32
+ path: dist/
33
+
34
+ publish-to-pypi:
35
+ name: >-
36
+ Publish Python 🐍 distribution 📦 to PyPI
37
+ if: startsWith(github.ref, 'refs/tags/') # only publish to PyPI on tag pushes
38
+ needs:
39
+ - build
40
+ runs-on: ubuntu-latest
41
+ environment:
42
+ name: pypi
43
+ url: https://pypi.org/p/MEDS-extract # Replace <package-name> with your PyPI project name
44
+ permissions:
45
+ id-token: write # IMPORTANT: mandatory for trusted publishing
46
+
47
+ steps:
48
+ - name: Download all the dists
49
+ uses: actions/download-artifact@v4
50
+ with:
51
+ name: python-package-distributions
52
+ path: dist/
53
+
54
+ - name: Publish distribution 📦 to PyPI
55
+ uses: pypa/gh-action-pypi-publish@release/v1
56
+
57
+ github-release:
58
+ name: >-
59
+ Sign the Python 🐍 distribution 📦 with Sigstore
60
+ and upload them to GitHub Release
61
+ needs:
62
+ - publish-to-pypi
63
+ runs-on: ubuntu-latest
64
+
65
+ permissions:
66
+ contents: write # IMPORTANT: mandatory for making GitHub Releases
67
+ id-token: write # IMPORTANT: mandatory for sigstore
68
+
69
+ steps:
70
+ - name: Download all the dists
71
+ uses: actions/download-artifact@v4
72
+ with:
73
+ name: python-package-distributions
74
+ path: dist/
75
+
76
+ - name: Sign the dists with Sigstore
77
+ uses: sigstore/gh-action-sigstore-python@v3.0.0
78
+ with:
79
+ inputs: >-
80
+ ./dist/*.tar.gz
81
+ ./dist/*.whl
82
+ - name: Create GitHub Release
83
+ env:
84
+ GITHUB_TOKEN: ${{ github.token }}
85
+ run: >-
86
+ gh release create
87
+ '${{ github.ref_name }}'
88
+ --repo '${{ github.repository }}'
89
+ --notes ""
90
+ - name: Upload artifact signatures to GitHub Release
91
+ env:
92
+ GITHUB_TOKEN: ${{ github.token }}
93
+ # Upload to GitHub Release using the `gh` CLI.
94
+ # `dist/` contains the built packages, and the
95
+ # sigstore-produced signatures and certificates.
96
+ run: >-
97
+ gh release upload
98
+ '${{ github.ref_name }}' dist/**
99
+ --repo '${{ github.repository }}'
@@ -0,0 +1,48 @@
1
+ name: Tests
2
+
3
+ on:
4
+ push:
5
+ branches: [main]
6
+ pull_request:
7
+ branches: [main, "release/*", "dev"]
8
+
9
+ jobs:
10
+ run_tests_ubuntu:
11
+ runs-on: ubuntu-latest
12
+
13
+ strategy:
14
+ matrix:
15
+ python-version: ["3.11", "3.12"]
16
+ fail-fast: false
17
+
18
+ timeout-minutes: 30
19
+
20
+ steps:
21
+ - name: Checkout
22
+ uses: actions/checkout@v4
23
+
24
+ - name: Set up Python ${{ matrix.python-version }}
25
+ uses: actions/setup-python@v5
26
+ with:
27
+ python-version: ${{ matrix.python-version }}
28
+
29
+ - name: Install packages
30
+ run: |
31
+ pip install -e .[tests]
32
+
33
+ #----------------------------------------------
34
+ # run test suite
35
+ #----------------------------------------------
36
+ - name: Run tests
37
+ run: |
38
+ pytest src/ tests/ -v --doctest-modules --cov=src --junitxml=junit.xml -s
39
+
40
+ - name: Upload coverage to Codecov
41
+ uses: codecov/codecov-action@v4.0.1
42
+ with:
43
+ token: ${{ secrets.CODECOV_TOKEN }}
44
+ - name: Upload test results to Codecov
45
+ if: ${{ !cancelled() }}
46
+ uses: codecov/test-results-action@v1
47
+ with:
48
+ token: ${{ secrets.CODECOV_TOKEN }}
@@ -0,0 +1,160 @@
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # C extensions
7
+ *.so
8
+
9
+ # Distribution / packaging
10
+ .Python
11
+ build/
12
+ develop-eggs/
13
+ dist/
14
+ downloads/
15
+ eggs/
16
+ .eggs/
17
+ lib/
18
+ lib64/
19
+ parts/
20
+ sdist/
21
+ var/
22
+ wheels/
23
+ share/python-wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .nox/
43
+ .coverage
44
+ .coverage.*
45
+ .cache
46
+ nosetests.xml
47
+ coverage.xml
48
+ *.cover
49
+ *.py,cover
50
+ .hypothesis/
51
+ .pytest_cache/
52
+ cover/
53
+
54
+ # Translations
55
+ *.mo
56
+ *.pot
57
+
58
+ # Django stuff:
59
+ *.log
60
+ local_settings.py
61
+ db.sqlite3
62
+ db.sqlite3-journal
63
+
64
+ # Flask stuff:
65
+ instance/
66
+ .webassets-cache
67
+
68
+ # Scrapy stuff:
69
+ .scrapy
70
+
71
+ # Sphinx documentation
72
+ docs/_build/
73
+
74
+ # PyBuilder
75
+ .pybuilder/
76
+ target/
77
+
78
+ # Jupyter Notebook
79
+ .ipynb_checkpoints
80
+
81
+ # IPython
82
+ profile_default/
83
+ ipython_config.py
84
+
85
+ # pyenv
86
+ # For a library or package, you might want to ignore these files since the code is
87
+ # intended to run in multiple environments; otherwise, check them in:
88
+ # .python-version
89
+
90
+ # pipenv
91
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
93
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
94
+ # install all needed dependencies.
95
+ #Pipfile.lock
96
+
97
+ # poetry
98
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
99
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
100
+ # commonly ignored for libraries.
101
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
102
+ #poetry.lock
103
+
104
+ # pdm
105
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
106
+ #pdm.lock
107
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
108
+ # in version control.
109
+ # https://pdm.fming.dev/#use-with-ide
110
+ .pdm.toml
111
+
112
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
113
+ __pypackages__/
114
+
115
+ # Celery stuff
116
+ celerybeat-schedule
117
+ celerybeat.pid
118
+
119
+ # SageMath parsed files
120
+ *.sage.py
121
+
122
+ # Environments
123
+ .env
124
+ .venv
125
+ env/
126
+ venv/
127
+ ENV/
128
+ env.bak/
129
+ venv.bak/
130
+
131
+ # Spyder project settings
132
+ .spyderproject
133
+ .spyproject
134
+
135
+ # Rope project settings
136
+ .ropeproject
137
+
138
+ # mkdocs documentation
139
+ /site
140
+
141
+ # mypy
142
+ .mypy_cache/
143
+ .dmypy.json
144
+ dmypy.json
145
+
146
+ # Pyre type checker
147
+ .pyre/
148
+
149
+ # pytype static type analyzer
150
+ .pytype/
151
+
152
+ # Cython debug symbols
153
+ cython_debug/
154
+
155
+ # PyCharm
156
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
157
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
158
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
159
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
160
+ #.idea/
@@ -0,0 +1,133 @@
1
+ default_language_version:
2
+ python: python3.12
3
+
4
+ exclude: "docs/index.md|MIMIC-IV_Example/README.md|eICU_Example/README.md|AUMCdb_Example/README.md"
5
+
6
+ repos:
7
+ - repo: https://github.com/pre-commit/pre-commit-hooks
8
+ rev: v5.0.0
9
+ hooks:
10
+ # list of supported hooks: https://pre-commit.com/hooks.html
11
+ - id: trailing-whitespace
12
+ - id: end-of-file-fixer
13
+ - id: check-docstring-first
14
+ - id: check-yaml
15
+ - id: debug-statements
16
+ - id: detect-private-key
17
+ - id: check-executables-have-shebangs
18
+ - id: check-toml
19
+ - id: check-case-conflict
20
+ - id: check-added-large-files
21
+ args: [--maxkb, "800"]
22
+
23
+ # python code formatting
24
+ - repo: https://github.com/psf/black
25
+ rev: 24.10.0
26
+ hooks:
27
+ - id: black
28
+ args: [--line-length, "110"]
29
+
30
+ # python import sorting
31
+ - repo: https://github.com/PyCQA/isort
32
+ rev: 5.13.2
33
+ hooks:
34
+ - id: isort
35
+ args: ["--profile", "black", "--filter-files", "-o", "wandb"]
36
+
37
+ - repo: https://github.com/PyCQA/autoflake
38
+ rev: v2.3.1
39
+ hooks:
40
+ - id: autoflake
41
+ args: [--in-place, --remove-all-unused-imports]
42
+
43
+ # python upgrading syntax to newer version
44
+ - repo: https://github.com/asottile/pyupgrade
45
+ rev: v3.19.0
46
+ hooks:
47
+ - id: pyupgrade
48
+ args: [--py311-plus]
49
+
50
+ # python docstring formatting
51
+ - repo: https://github.com/myint/docformatter
52
+ rev: v1.7.5
53
+ hooks:
54
+ - id: docformatter
55
+ args: [--in-place, --wrap-summaries=110, --wrap-descriptions=110]
56
+
57
+ # python check (PEP8), programming errors and code complexity
58
+ - repo: https://github.com/PyCQA/flake8
59
+ rev: 7.1.1
60
+ hooks:
61
+ - id: flake8
62
+ args:
63
+ [
64
+ "--max-complexity=10",
65
+ "--extend-ignore",
66
+ "E402,E701,E251,E226,E302,W504,E704,E402,E401,C901,E203",
67
+ "--max-line-length=110",
68
+ "--exclude",
69
+ "logs/*,data/*",
70
+ "--per-file-ignores",
71
+ "__init__.py:F401",
72
+ ]
73
+
74
+ # yaml formatting
75
+ - repo: https://github.com/pre-commit/mirrors-prettier
76
+ rev: v4.0.0-alpha.8
77
+ hooks:
78
+ - id: prettier
79
+ types: [yaml]
80
+ exclude: "environment.yaml"
81
+
82
+ # shell scripts linter
83
+ - repo: https://github.com/shellcheck-py/shellcheck-py
84
+ rev: v0.10.0.1
85
+ hooks:
86
+ - id: shellcheck
87
+
88
+ # md formatting
89
+ - repo: https://github.com/executablebooks/mdformat
90
+ rev: 0.7.18
91
+ hooks:
92
+ - id: mdformat
93
+ args: ["--number"]
94
+ additional_dependencies:
95
+ - mdformat-gfm
96
+ - mdformat-tables
97
+ - mdformat_frontmatter
98
+ - mdformat-black
99
+ - mdformat-config
100
+ - mdformat-shfmt
101
+ - mdformat-mkdocs
102
+ - mdformat-toc
103
+ - mdformat-admon
104
+
105
+ # word spelling linter
106
+ - repo: https://github.com/codespell-project/codespell
107
+ rev: v2.3.0
108
+ hooks:
109
+ - id: codespell
110
+ args:
111
+ - --skip=logs/**,data/**,*.ipynb,*.bib,env.yml,env_cpu.yml,*.svg,poetry.lock
112
+ - --ignore-words-list=ehr,crate
113
+
114
+ # jupyter notebook cell output clearing
115
+ - repo: https://github.com/kynan/nbstripout
116
+ rev: 0.7.1
117
+ hooks:
118
+ - id: nbstripout
119
+
120
+ # jupyter notebook linting
121
+ - repo: https://github.com/nbQA-dev/nbQA
122
+ rev: 1.8.7
123
+ hooks:
124
+ - id: nbqa-black
125
+ args: ["--line-length=110"]
126
+ - id: nbqa-isort
127
+ args: ["--profile=black"]
128
+ - id: nbqa-flake8
129
+ args:
130
+ [
131
+ "--extend-ignore=E203,E402,E501,F401,F841",
132
+ "--exclude=logs/*,data/*",
133
+ ]
@@ -0,0 +1,21 @@
1
+ # Read the Docs configuration file
2
+ # See https://docs.readthedocs.io/en/stable/config-file/v2.html for details
3
+
4
+ # Required
5
+ version: 2
6
+
7
+ # Set the version of Python and other tools you might need
8
+ build:
9
+ os: ubuntu-22.04
10
+ tools:
11
+ python: "3.12"
12
+
13
+ python:
14
+ install:
15
+ - method: pip
16
+ path: .
17
+ extra_requirements:
18
+ - docs
19
+
20
+ mkdocs:
21
+ configuration: mkdocs.yml
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2024 Matthew McDermott
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.