fix bug

Author: zhiqiu

python/paddle/fluid/dataloader/dataloader_iter.py

@@ -303,7 +303,7 @@ def _init_thread(self):
         self._thread.daemon = True
         self._thread.start()
 
-    def _thread_loop(self, legacy_expected_place=None):
+    def _thread_loop(self, legacy_expected_place):
         try:
             #NOTE(zhiqiu): Set the expected place for new thread as the same as father thread,
             # and it will call platform::SetDeviceId() in c++ internally.
@@ -613,7 +613,7 @@ def _exit_thread_unexpectedly(self):
         self._blocking_queue.kill()
         logging.error("DataLoader reader thread raised an exception!")
 
-    def _thread_loop(self, legacy_expected_place=None):
+    def _thread_loop(self, legacy_expected_place):
         #NOTE(zhiqiu): Set the expected place for new thread as the same as father thread,
         # and it will call platform::SetDeviceId() in c++ internally.
         # If we do not set cudaDeviceId in new thread, the default cudaDeviceId will be 0,

python/paddle/fluid/reader.py

@@ -971,7 +971,7 @@ def _exit_thread_unexpectedly(self):
         self._blocking_queue.kill()
         logging.error("DataLoader reader thread raised an exception!")
 
-    def _reader_thread_loop_for_multiprocess(self):
+    def _reader_thread_loop_for_multiprocess(self, legacy_expected_place):
         # See _DataLoaderIterSingleProcess._thread_loop() for why set expected place here.
         _set_expected_place(legacy_expected_place)
 
@@ -1013,7 +1013,7 @@ def _reader_thread_loop_for_multiprocess(self):
                 else:
                     self._exit_thread_expectedly()
 
-    def _reader_thread_loop_for_singleprocess(self):
+    def _reader_thread_loop_for_singleprocess(self, legacy_expected_place):
         try:
             # See _DataLoaderIterSingleProcess._thread_loop() for why set expected place here.
             _set_expected_place(legacy_expected_place)