Commit 8a0823eb authored by Thomas Purcell's avatar Thomas Purcell
Browse files

Fix type for initalize arrays

parent 7aec32ff
......@@ -36,7 +36,12 @@ def test_param_add_node():
task_sizes_train = [90]
task_sizes_test = [10]
initialize_values_arr(np.array(task_sizes_train), np.array(task_sizes_test), 2, 1)
initialize_values_arr(
np.array(task_sizes_train, dtype=np.int32),
np.array(task_sizes_test, dtype=np.int32),
2,
1,
)
initialize_param_storage()
data_1 = np.linspace(-20, 20, task_sizes_train[0])
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment