nip.utils.types

nip.utils.types#

Type definitions for the NIP package.

Module Attributes

A nested key type used for tensordict keys.

Functions

get_union_elements(tp)

Get the elements of a union type.

Classes

`DpoDatasetItem`	A single item in a DPO dataset.
`ExperimentConfig`	Configuration for an experiment, which specifies the hyperparameters.
`HuggingFaceDpoDatasetItem`	A single item in a DPO dataset using the Hugging Face format.
`PromptMessage`	A message in the prompt for a language model API.
`SupervisedDatasetItem`	A single item in a supervised dataset.