ChoiceParameter with list of ints: TypeError: Object of type int32 is not JSON serializable

[facebook]

ChoiceParameter with list of ints: TypeError: Object of type int32 is not JSON serializable

[ldworkin]

Hi @BCJuan ! Thanks for reporting this issue. I'm having trouble reproducing though. This is what I ran:

from ax import *

    range_x = ChoiceParameter(name='x', values=[1,5,10],
                              parameter_type=ParameterType.INT)
    space = SearchSpace(parameters=[range_x])
    experiment = Experiment(name="experiment_one_cell",
                            search_space=space)
    sobol = Models.SOBOL(search_space=experiment.search_space)
    generator_run = sobol.gen(100)

However, that completed successfully. Am I missing something?

[BCJuan]

Hi,

I still havethe problem if I use your code. It is my fault, I have checked the datatype and mine are numpy.int32

The function numpy_type_to_python_type in typeutils.py only catches np.int64 and numpy.float64.

I suppose is mandatory to use 64 bit data types.

Apologies for the time spent.

Best

[ldworkin]

Oh interesting! We can fix it to work with np.int32 and np.float32. Thanks for catching this!

[ldworkin]

Actually I believe this was already fixed! What version of Ax are you using?

see:
7904aa2#diff-0befb5b9d1a4f64449eba6e50b2d2b0d

[BCJuan]

My fault I am using 0.1.3. Closing the issue. Thanks!

[ldworkin]

Great!