Skip to content

Commit 7b949c3

Browse files
authored
Merge pull request #20 from FGRCL/fix-batch-test
Fix batch unit test
2 parents 91dab28 + f2a7149 commit 7b949c3

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

src/tfdatacompose/batch.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from tensorflow.python.data import AUTOTUNE, Dataset
1+
from tensorflow.python.data import Dataset
22

33
from tfdatacompose.datasetoperation import DatasetOperation
44

@@ -24,7 +24,5 @@ def apply(self, dataset: Dataset) -> Dataset:
2424
return dataset.batch(
2525
self.batch_size,
2626
self.drop_remainder,
27-
num_parallel_calls=AUTOTUNE,
28-
deterministic=False,
2927
name=self.__class__.__name__,
3028
)

0 commit comments

Comments
 (0)