|
@@ -88,7 +88,7 @@ class TSBatchSampler(BaseBatchSampler):
|
|
|
batch.append(input, None)
|
|
batch.append(input, None)
|
|
|
if len(batch) == self.batch_size:
|
|
if len(batch) == self.batch_size:
|
|
|
yield batch
|
|
yield batch
|
|
|
- batch.reset()
|
|
|
|
|
|
|
+ batch = Batch()
|
|
|
elif isinstance(input, str):
|
|
elif isinstance(input, str):
|
|
|
file_path = (
|
|
file_path = (
|
|
|
self._download_from_url(input)
|
|
self._download_from_url(input)
|
|
@@ -100,7 +100,7 @@ class TSBatchSampler(BaseBatchSampler):
|
|
|
batch.append(file_path, file_path)
|
|
batch.append(file_path, file_path)
|
|
|
if len(batch) == self.batch_size:
|
|
if len(batch) == self.batch_size:
|
|
|
yield batch
|
|
yield batch
|
|
|
- batch.reset()
|
|
|
|
|
|
|
+ batch = Batch()
|
|
|
else:
|
|
else:
|
|
|
logging.warning(
|
|
logging.warning(
|
|
|
f"Not supported input data type! Only `pd.DataFrame` and `str` are supported! So has been ignored: {input}."
|
|
f"Not supported input data type! Only `pd.DataFrame` and `str` are supported! So has been ignored: {input}."
|