We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e48df03 commit 92a5673Copy full SHA for 92a5673
tests/test_approximators/test_estimate.py
@@ -9,6 +9,7 @@ def test_approximator_estimate(approximator, simulator, batch_size, adapter):
9
data = simulator.sample((num_batches * batch_size,))
10
11
batch = adapter(data)
12
+ batch = keras.tree.map_structure(keras.ops.convert_to_tensor, batch)
13
approximator.build_from_data(batch)
14
15
estimates = approximator.estimate(data)
0 commit comments