onmep 0.1.0__tar.gz

onmep-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 hbmep
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

onmep-0.1.0/PKG-INFO

@@ -0,0 +1,98 @@
+Metadata-Version: 2.4
+Name: onmep
+Version: 0.1.0
+Summary: Online adaptive acquisition of motor-evoked potential recruitment curves
+Author-email: Vishweshwar Tyagi <vt2353@columbia.edu>, "James R. McIntosh" <jrm2263@cumc.columbia.edu>
+License: MIT License
+        
+        Copyright (c) 2026 hbmep
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
+Project-URL: Homepage, https://github.com/hbmep/onmep
+Project-URL: Bug Tracker, https://github.com/hbmep/onmep
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development
+Classifier: Topic :: Scientific/Engineering
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX
+Classifier: Operating System :: Unix
+Classifier: Operating System :: MacOS
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: hbmep>=0.8.0
+Requires-Dist: numpy<2.0,>=1.24.2
+Requires-Dist: jax>=0.4.24
+Requires-Dist: numpyro>=0.13.2
+Requires-Dist: pandas>=2.0.2
+Requires-Dist: scikit-learn>=1.2.1
+Requires-Dist: arviz>=0.18.0
+Requires-Dist: matplotlib>=3.7.0
+Requires-Dist: seaborn>=0.12.2
+Requires-Dist: joblib>=1.3.2
+Requires-Dist: pylsl>=1.18.1
+Requires-Dist: xmltodict>=1.0.2
+Requires-Dist: pydantic>=2.13.1
+Provides-Extra: dev
+Requires-Dist: notebook>=7.3.2; extra == "dev"
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: pytest-pycodestyle>=2.3.1; extra == "dev"
+Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
+Requires-Dist: Sphinx==7.3.7; extra == "dev"
+Requires-Dist: sphinx-autobuild==2024.04.16; extra == "dev"
+Requires-Dist: sphinx-book-theme==1.1.3; extra == "dev"
+Requires-Dist: sphinxcontrib.bibtex==2.6.2; extra == "dev"
+Requires-Dist: sphinx-copybutton==0.5.2; extra == "dev"
+Requires-Dist: myst_nb==1.1.0; extra == "dev"
+Provides-Extra: cuda12
+Requires-Dist: jax[cuda12]>=0.4.24; extra == "cuda12"
+Dynamic: license-file
+
+# onMEP
+
+onMEP is a Python library for online adaptive acquisition of motor-evoked potential (MEP) recruitment curves.
+
+[![Cite](https://img.shields.io/badge/DOI-10.1016%2Fj.brs.2025.09.008-blue.svg)](https://doi.org/10.1016/j.brs.2025.09.008)
+[![Cite](https://img.shields.io/badge/stats.AP-arXiv:2407.08709-B31B1B.svg)](http://doi.org/10.48550/arXiv.2407.08709)
+
+## Citation
+
+Please cite [Tyagi et al., 2025](https://doi.org/10.1016/j.brs.2025.09.008) if you find this code useful in your research. The BibTeX entry for the paper is:
+
+```
+@article{tyagi_hierarchical_2025,
+    title = {Hierarchical {Bayesian} estimation of motor-evoked potential recruitment curves yields accurate and robust estimates},
+    author = {Tyagi, Vishweshwar and Murray, Lynda M. and Asan, Ahmet S. and Mandigo, Christopher and Virk, Michael S. and Harel, Noam Y. and Carmel, Jason B. and McIntosh, James R.},
+    journal = {Brain Stimulation},
+    year = {2025},
+    doi = {10.1016/j.brs.2025.09.008},
+}
+```
+
+## License
+
+onMEP is free software made available under the MIT License. For details see the LICENSE file.

onmep-0.1.0/README.md

@@ -0,0 +1,24 @@
+# onMEP
+
+onMEP is a Python library for online adaptive acquisition of motor-evoked potential (MEP) recruitment curves.
+
+[![Cite](https://img.shields.io/badge/DOI-10.1016%2Fj.brs.2025.09.008-blue.svg)](https://doi.org/10.1016/j.brs.2025.09.008)
+[![Cite](https://img.shields.io/badge/stats.AP-arXiv:2407.08709-B31B1B.svg)](http://doi.org/10.48550/arXiv.2407.08709)
+
+## Citation
+
+Please cite [Tyagi et al., 2025](https://doi.org/10.1016/j.brs.2025.09.008) if you find this code useful in your research. The BibTeX entry for the paper is:
+
+```
+@article{tyagi_hierarchical_2025,
+    title = {Hierarchical {Bayesian} estimation of motor-evoked potential recruitment curves yields accurate and robust estimates},
+    author = {Tyagi, Vishweshwar and Murray, Lynda M. and Asan, Ahmet S. and Mandigo, Christopher and Virk, Michael S. and Harel, Noam Y. and Carmel, Jason B. and McIntosh, James R.},
+    journal = {Brain Stimulation},
+    year = {2025},
+    doi = {10.1016/j.brs.2025.09.008},
+}
+```
+
+## License
+
+onMEP is free software made available under the MIT License. For details see the LICENSE file.

onmep-0.1.0/pyproject.toml

@@ -0,0 +1,73 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "onmep"
+version = "0.1.0"
+description = "Online adaptive acquisition of motor-evoked potential recruitment curves"
+readme = { file = "README.md", content-type = "text/markdown" }
+requires-python = ">=3.11"
+license = { file = "LICENSE" }
+authors = [
+  { name = "Vishweshwar Tyagi", email = "vt2353@columbia.edu" },
+  { name = "James R. McIntosh", email = "jrm2263@cumc.columbia.edu" }
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Science/Research",
+    "Intended Audience :: Developers",
+    "Topic :: Software Development",
+    "Topic :: Scientific/Engineering",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3 :: Only",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: Microsoft :: Windows",
+    "Operating System :: POSIX",
+    "Operating System :: Unix",
+    "Operating System :: MacOS"
+]
+dependencies = [
+    "hbmep>=0.8.0",
+    "numpy>=1.24.2,<2.0",
+    "jax>=0.4.24",
+    "numpyro>=0.13.2",
+    "pandas>=2.0.2",
+    "scikit-learn>=1.2.1",
+    "arviz>=0.18.0",
+    "matplotlib>=3.7.0",
+    "seaborn>=0.12.2",
+    "joblib>=1.3.2",
+    "pylsl>=1.18.1",
+    "xmltodict>=1.0.2",
+    "pydantic>=2.13.1",
+]
+
+[project.optional-dependencies]
+dev = [
+    "notebook>=7.3.2",
+    # Testing
+    "pytest>=7.4.0",
+    "pytest-pycodestyle>=2.3.1",
+    "pytest-cov>=4.1.0",
+    # Docs
+    "Sphinx==7.3.7",
+    "sphinx-autobuild==2024.04.16",
+    "sphinx-book-theme==1.1.3",
+    "sphinxcontrib.bibtex==2.6.2",
+    "sphinx-copybutton==0.5.2",
+    "myst_nb==1.1.0"
+]
+cuda12 = ["jax[cuda12]>=0.4.24"]
+
+[tool.pytest.ini_options]
+minversion = "6.0"
+addopts = "-v --color=yes --junit-xml=out/junit/junit.xml --pycodestyle src --cov --cov-report term-missing --doctest-modules"
+junit_family = "xunit2"
+testpaths = ["tests"]
+
+[project.urls]
+"Homepage" = "https://github.com/hbmep/onmep"
+"Bug Tracker" = "https://github.com/hbmep/onmep"

onmep-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

onmep-0.1.0/src/onmep/__init__.py

@@ -0,0 +1,8 @@
+import hbmep
+
+from importlib.metadata import version
+__version__ = version("onmep")
+
+from onmep.checkpoint import CheckpointManager
+from onmep.structures import Observation, ObsArrays
+from onmep.telemetry import _now_ns, StepTimes, Telemetry

onmep-0.1.0/src/onmep/agent/__init__.py

@@ -0,0 +1,5 @@
+from onmep.agent.base_agent import BaseAgent
+
+__all__ = [
+    "BaseAgent",
+]