fix codecheck

This commit is contained in:
liyong 2021-06-07 09:57:25 +08:00
parent a4e2ab0487
commit a366798555
2 changed files with 6 additions and 0 deletions

View File

@ -3007,6 +3007,9 @@ class TransferDataset(Dataset):
self._to_device.continue_send()
def get_data_info(self):
"""
Get type and shape of current batch
"""
if self._to_device is not None:
return self._to_device.get_data_info()
raise RuntimeError("Calling get_data_info with bad state.")

View File

@ -74,6 +74,9 @@ class CsvToMR:
@staticmethod
def check_columns(columns, columns_name):
"""
Validate the columns of csv
"""
if not columns:
return
if isinstance(columns, list):