PRIOR

🐍 A Python package for distributing resources from AI2's PRIOR team

Installation

Install the prior package with pip:

pip install prior

Datasets

• ProcTHOR-10k [GitHub]

import prior
prior.load_dataset("procthor-10k")

• Object Nav Evaluation [GitHub]

import prior
prior.load_dataset("object-nav-eval")

Models

• ProcTHOR Models [GitHub]

import prior
prior.load_model(project="procthor-models", model="object-nav-pretraining")

Example Usage

To use a public Python dataset, simply run:

import prior
dataset = prior.load_dataset("test-dataset", entity="mattdeitke", revision="main")

Here, revision can be either a tag, branch, or commit hash.

Private Datasets

If you want to use a private dataset, make sure you're either:

1. Already logged into GitHub from the command line, and able to pull a private repo.
2. Set the GITHUB_TOKEN environment variable to a GitHub authentication token with read access to private repositories (e.g., export GITHUB_TOKEN=<token>). You can generate a GitHub authentication token here.
3. Set the gh_auth_token global variable in the prior package with:

import prior
prior.gh_auth_token = "<token>"