Apply isort and black reformatting

skothenhill-nv · skothenhill-nv · commit 2d472cc2484f · 2024-06-14T16:42:29.000Z
Signed-off-by: skothenhill-nv &lt;skothenhill-nv@users.noreply.github.com&gt;
diff --git a/nemo/lightning/pytorch/strategies.py b/nemo/lightning/pytorch/strategies.py
@@ -508,7 +508,7 @@ def parallelism(self):
             tensor_model_parallel_size=self.tensor_model_parallel_size,
             pipeline_model_parallel_size=self.pipeline_model_parallel_size,
             virtual_pipeline_model_parallel_size=self.virtual_pipeline_model_parallel_size,
-            pipeline_dtype=self.pipeline_dtype
+            pipeline_dtype=self.pipeline_dtype,
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -508,7 +508,7 @@ def parallelism(self):`
`508`	`508`	`tensor_model_parallel_size=self.tensor_model_parallel_size,`
`509`	`509`	`pipeline_model_parallel_size=self.pipeline_model_parallel_size,`
`510`	`510`	`virtual_pipeline_model_parallel_size=self.virtual_pipeline_model_parallel_size,`
`511`		`- pipeline_dtype=self.pipeline_dtype`
	`511`	`+ pipeline_dtype=self.pipeline_dtype,`
`512`	`512`	`)`
`513`	`513`
`514`	`514`