Skip to content

Commit beced88

Browse files
committed
fix unittest. test=develop
1 parent 1c21737 commit beced88

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/fluid/tests/unittests/test_multiprocess_dataloader_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -338,7 +338,7 @@ def setUp(self):
338338
[457190280, 2660306227, 859341110, 354512857]]
339339

340340
def test_main(self):
341-
from paddle.fluid.dataloader.state import _generate_states
341+
from paddle.fluid.dataloader.worker import _generate_states
342342
for inp, outp in zip(self.inputs, self.outputs):
343343
out = _generate_states(*inp)
344344
assert out == outp

0 commit comments

Comments
 (0)