Skip to content

Commit 92a5673

Browse files
committed
fix conversion to numpy in test_approximator_estimate
1 parent e48df03 commit 92a5673

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_approximators/test_estimate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ def test_approximator_estimate(approximator, simulator, batch_size, adapter):
99
data = simulator.sample((num_batches * batch_size,))
1010

1111
batch = adapter(data)
12+
batch = keras.tree.map_structure(keras.ops.convert_to_tensor, batch)
1213
approximator.build_from_data(batch)
1314

1415
estimates = approximator.estimate(data)

0 commit comments

Comments
 (0)