Metadata-Version: 2.4
Name: rueda
Version: 0.0.1.dev1
Summary: Rueda — Physical AI Data Platform
Author-email: Vistralis Labs <support@vistralis.org>
License-Expression: Apache-2.0
Project-URL: Homepage, https://vistralis.org/rueda
Project-URL: Documentation, https://vistralis.org/rueda/docs
Project-URL: Source, https://github.com/vistralis/rueda
Project-URL: Issues, https://github.com/vistralis/rueda/issues
Keywords: datasets,robotics,physical-ai,autonomous-vehicles,machine-learning,computer-vision
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: numpy>=2.0.0
Requires-Dist: numpy-quaternion
Requires-Dist: networkx
Requires-Dist: Pillow>=10.0
Requires-Dist: pyyaml>=6.0
Requires-Dist: tqdm>=4.65
Requires-Dist: transform-graph>=0.1.2
Provides-Extra: viz
Requires-Dist: plotly>=5.0; extra == "viz"
Provides-Extra: argoverse
Requires-Dist: pandas; extra == "argoverse"
Requires-Dist: pyarrow; extra == "argoverse"
Requires-Dist: av2; extra == "argoverse"
Provides-Extra: dev
Requires-Dist: pytest>=8.0; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Provides-Extra: all
Requires-Dist: rueda[dev,viz]; extra == "all"
Dynamic: license-file

# rueda

![CI](https://github.com/vistralis/rueda/actions/workflows/ci.yml/badge.svg)
[![PyPI](https://img.shields.io/pypi/v/rueda)](https://pypi.org/project/rueda/)
[![License](https://img.shields.io/badge/license-Apache%202.0-blue)](LICENSE)

**Physical AI Data Platform — unified multi-sensor dataset access, spatially-aware augmentation, and task-specific tensor generation for robotics and autonomous driving.**

## Installation

```bash
pip install rueda
```

For visualization support:
```bash
pip install "rueda[viz]"
```

## Overview

Rueda provides a three-layer pipeline for Physical AI data:

1. **WorldDataset** — Format-agnostic access to multi-sensor datasets (NuScenes, KITTI, Argoverse2, TUMTraf). Each sample bundles camera images, LiDAR pointclouds, calibration, poses, and 3D annotations into a unified `DataSample`.

2. **Augmenters** — Spatially-aware augmentation that respects sensor geometry. Transforms are applied consistently across all modalities using `transform-graph` for coordinate frame management.

3. **Generators** — Task-specific tensor generation. Convert augmented `DataSample` instances into model-ready tensors (BEV grids, voxel occupancy, VLM context).

## Architecture

```
Raw Data → WorldDataset → DataSample → Augmenter → DataSample → Generator → Tensors
                                                                      ↓
                                                              ReasoningDataset
                                                                      ↓
                                                               DataContext → VLM
```

### Reasoning Pipeline

For Vision-Language Models, `ReasoningDataset` wraps any `WorldDataset` (or loads lightweight image/video/text datasets directly) to produce `DataContext` — a structured container with frames, prompts, and ground truth ready for VLM consumption.

## Documentation

Full documentation is available at [vistralis.org/rueda](https://vistralis.org/rueda).

## Development

Install in editable mode with development dependencies:
```bash
pip install -e ".[dev]"
```

### Linting & Formatting

```bash
ruff check .
ruff format .
```

### Testing

```bash
pytest
pytest --cov=rueda --cov-report=term-missing
```

## CI/CD

- **CI**: Lint + test on Python 3.10–3.14 (push to main, PRs)
- **Release**: Tag `v*` triggers automated PyPI publish via trusted publishing

## License

Apache 2.0 — Vistralis Labs
