`gfn.containers.base`

Module Contents

Base class for states containers (states, transitions, or trajectories)

class gfn.containers.base.Container

Bases: abc.ABC

Base class for states containers (states, transitions, or trajectories)

abstract __getitem__(index)

Subsets the container

abstract __len__()

Returns the number of elements in the container

abstract extend(other)

Extends the current container

load(path)

Loads the container from a file, overwriting the current container

sample(n_samples)

Samples a subset of the container

save(path)

Saves the container to a file