chore: Linter fixes

Signed-off-by: Dheeraj Peri <peri.dheeraj@gmail.com>
pytorch · Apr 19, 2023 · 2addf5e · 2addf5e
1 parent 13caff0
commit 2addf5e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/py/torch_tensorrt/_Input.py b/py/torch_tensorrt/_Input.py
@@ -384,9 +384,7 @@ def example_tensor(self, optimization_profile_field: str = None) -> torch.Tensor
             )
 
         if self.shape_mode == Input._ShapeMode.STATIC:
-            return torch.rand(self.shape).to(
-                dtype=self.torch_dtype
-            )
+            return torch.rand(self.shape).to(dtype=self.torch_dtype)
         else:
             return torch.rand(self.shape[optimization_profile_field]).to(
                 dtype=self.torch_dtype

diff --git a/py/torch_tensorrt/dynamo/fx_ts_compat/test/core/test_input_tensor_spec.py b/py/torch_tensorrt/dynamo/fx_ts_compat/test/core/test_input_tensor_spec.py
@@ -72,7 +72,9 @@ def test_from_static_input(self):
             self._validate_spec(spec, tensor)
 
     def test_from_dynamic_input(self):
-        inputs = torch_tensorrt.Input(min_shape=(2, 2, 3), opt_shape=(4, 2, 3), max_shape=(8, 2, 3))
+        inputs = torch_tensorrt.Input(
+            min_shape=(2, 2, 3), opt_shape=(4, 2, 3), max_shape=(8, 2, 3)
+        )
         example_tensor = inputs.example_tensor(optimization_profile_field="opt_shape")
         spec = InputTensorSpec.from_input(inputs)
         self._validate_spec(spec, example_tensor, dynamic_dims=[0])