Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -10,4 +10,5 @@ __pycache__/*
*~
*.swp
*.swo
*.swn
stats.*.pickle
1 change: 1 addition & 0 deletions .python-version
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
3.13
49 changes: 41 additions & 8 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,15 +28,48 @@ It combines classic Computer Vision heuristics (blur / exposure tests, kinematic

## ⚙️ Installation

```bash
# clone your repo first
pip install -r requirements.txt
export GOOGLE_API_KEY="sk-..." # Required only if you use VLM-based scoring

## Example
HF_USER='...'
python score_dataset.py --dataset /path/to/data/${HF_USER}/open-book --task "Open the book"
### Installation
```
git clone git@github.com:RoboticsData/score_lerobot_episodes.git
cd score_lerobot_episodes

uv venv
source .venv/bin/activate
uv sync # installs dependencies expressed in pyproject.toml and uv.lock

python -c 'import score_dataset' || echo 'Something is wrong, check dependencies'

# Test if you're logged into the hugginface CLI:
hf auth whoami
# If not, go to https://huggingface.co/settings/tokens, make a new token, and login:
hf auth login
```

### Usage

```
# in the score_lerobot_episodes directory:

source .venv/bin/activate

# Fetch a dataset from HuggingFace, example:
PATH_TO_HF_DATASET=`hf download Rorschach4153/so101_60_new --repo-type dataset`

## TODO make tool injest HF repo names instead of local paths to cached snapshots
python score_dataset.py --dataset ${PATH_TO_HF_DATASET} --task "none" --camera 'phone'
```

### Adding dependencies

Use ```uv add``` to add dependencies, then ensure you commit changes to pyproject.toml and uv.lock.

#### Python versions

We opt for keeping ```.python-version``` in the repo as a way to interlock the Python
runtime version with installed dependecy versions. Python packages express
dependence on difference Python versions, so Python itself effectively becomes
a dependency.


## Star History

Expand Down
16 changes: 16 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
[project]
name = "score-lerobot-episodes"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.13"
dependencies = [
"dotenv>=0.9.9",
"google-generativeai>=0.8.5",
"huggingface-hub[cli]>=0.34.3",
"numpy>=2.3.2",
"opencv-python>=4.11.0.86",
"pandas>=2.3.1",
"pyarrow>=21.0.0",
"uniplot>=0.21.3",
]
5 changes: 0 additions & 5 deletions requirements.txt

This file was deleted.

Loading