gflownet.envs.composite.utils

Functions

make_set(is_flexible[, subenvs, max_elements, ...])

Factory method to create SetFix or SetFlex classes depending on the input

Module Contents

gflownet.envs.composite.utils.make_set(is_flexible, subenvs=None, max_elements=None, envs_unique=None, do_random_subenvs=False, **kwargs)

Factory method to create SetFix or SetFlex classes depending on the input is_flexible.

This method mimics conditional inheritance.

Parameters:

• is_flexible (bool) – If True, return a SetFlex environment. If False, return a SetFix environment.

• subenvs (Iterable[gflownet.envs.base.GFlowNetEnv])

• max_elements (int)

• envs_unique (Iterable[gflownet.envs.base.GFlowNetEnv])

• do_random_subenvs (bool)