Metadata-Version: 2.4
Name: hemosim
Version: 0.1.0
Summary: Reinforcement learning Gymnasium environments for hemostasis and anticoagulation management
Author-email: Hass Dhia <partners@smarttechinvest.com>
License-Expression: MIT
License-File: LICENSE
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Medical Science Apps.
Requires-Python: >=3.10
Requires-Dist: gymnasium>=0.29
Requires-Dist: numpy>=1.24
Requires-Dist: scipy>=1.11
Provides-Extra: all
Requires-Dist: mypy>=1.0; extra == 'all'
Requires-Dist: pytest-cov>=4.0; extra == 'all'
Requires-Dist: pytest>=7.0; extra == 'all'
Requires-Dist: ruff>=0.1; extra == 'all'
Requires-Dist: stable-baselines3>=2.0; extra == 'all'
Requires-Dist: torch>=2.0; extra == 'all'
Provides-Extra: dev
Requires-Dist: mypy>=1.0; extra == 'dev'
Requires-Dist: pytest-cov>=4.0; extra == 'dev'
Requires-Dist: pytest>=7.0; extra == 'dev'
Requires-Dist: ruff>=0.1; extra == 'dev'
Provides-Extra: train
Requires-Dist: stable-baselines3>=2.0; extra == 'train'
Requires-Dist: torch>=2.0; extra == 'train'
Description-Content-Type: text/markdown

# HemoSim

Reinforcement learning Gymnasium environments for hemostasis and anticoagulation management.

## Installation

```bash
pip install hemosim
```

## Environments

| Environment | Description | Observation | Action |
|---|---|---|---|
| `hemosim/WarfarinDosing-v0` | 90-day warfarin titration | Box(8) | Box(1) |
| `hemosim/HeparinInfusion-v0` | 5-day heparin infusion | Box(6) | Box(2) |
| `hemosim/DOACManagement-v0` | 365-day DOAC selection | Box(8) | MultiDiscrete(3,3) |
| `hemosim/DICManagement-v0` | 7-day DIC management | Box(8) | MultiDiscrete(4,4,3,3) |

## Quick Start

```python
import gymnasium as gym
import hemosim

env = gym.make("hemosim/WarfarinDosing-v0")
obs, info = env.reset(seed=42)

for _ in range(90):
    action = env.action_space.sample()
    obs, reward, terminated, truncated, info = env.step(action)
    if terminated or truncated:
        break

env.close()
```

## License

MIT License. Copyright (c) 2026 Hass Dhia, Smart Technology Investments Research Institute.
