nip.scenario_base.rollout_samples.register_rollout_samples_class

nip.scenario_base.rollout_samples.register_rollout_samples_class#

nip.scenario_base.rollout_samples.register_rollout_samples_class(scenario: Literal['graph_isomorphism', 'image_classification', 'code_validation'])[source]#

Register a subclass of RolloutSamples with a scenario.

Parameters:

scenario (ScenarioType) – The scenario with which to register the subclass.