diff --git a/torch2trt/dataset.py b/torch2trt/dataset.py index 5d49cea2..3cf846cb 100644 --- a/torch2trt/dataset.py +++ b/torch2trt/dataset.py @@ -58,7 +58,7 @@ def num_inputs(self): @property def flattener(self): if not hasattr(self, '_flattener') or self._flattener is None: - assert(len(self) > 0, 'Cannot create default flattener without input data.') + assert len(self) > 0, 'Cannot create default flattener without input data.' value = self[0] self._flattener = Flattener.from_value(value) return self._flattener