Clean up

Signed-off-by: ksztenderski <ksztenderski@nvidia.com>
NVIDIA · Apr 1, 2022 · b3846c4 · b3846c4
1 parent 8ff9509
commit b3846c4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dali/python/nvidia/dali/_debug_mode.py b/dali/python/nvidia/dali/_debug_mode.py
@@ -356,8 +356,7 @@ def __init__(self, op_class, op_name, next_logical_id, batch_size, seed, inputs,
         self.op_helper = op_class(**self._init_args)
         self._op_name = op_name
         self.op_spec = self.op_helper._spec
-        self.logical_ids = [id for id in range(
-            next_logical_id, next_logical_id + abs(input_set_len))]
+        self.logical_ids = [id for id in range(next_logical_id, next_logical_id + abs(input_set_len))]
 
         for i in range(len(inputs)):
             self.op_spec.AddInput(op_name+f'[{i}]', self._inputs_classification[i].device)