Wrapper bug when no state variable is used

[musslick]

This is will throw an error (since none of the standard state fields are used).

@on_state()
def random_sample_on_state(all_conditions, num_samples, random_state=None):
  return Delta(conditions=random_sample(all_conditions, num_samples, random_state))

and currently requires this workaround:

@on_state()
def random_sample_on_state(conditions, all_conditions, num_samples, random_state=None):
  return Delta(conditions=random_sample(all_conditions, num_samples, random_state))

[younesStrittmatter]

Could not reproduce the error. Maybe you forgot the keyword argument or to pass in the state. This works:

import pandas as pd

from autora.experimentalist.random import random_sample
from autora.state import StandardState, on_state, Delta


@on_state
def random_sample_on_state(all_conditions):
    return Delta(conditions=random_sample(all_conditions, 1, 42))


all_conditions = pd.DataFrame({'x': [0, 1, 2]})
s = StandardState()
s = random_sample_on_state(s, all_conditions=all_conditions)
print(s.conditions)