fix lint

huggingface · Oct 31, 2023 · aa4f49e · aa4f49e
1 parent 5a55117
commit aa4f49e
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/datatrove/executor/slurm.py b/src/datatrove/executor/slurm.py
@@ -160,12 +160,16 @@ def launch_file(self):
     def get_launch_file(self, sbatch_args: dict, run_script: str):
         args = "\n".join([f"#SBATCH --{k}={v}" for k, v in sbatch_args.items()])
 
-        env_command = self.env_command if self.env_command else (
-            f"""conda init bash
+        env_command = (
+            self.env_command
+            if self.env_command
+            else (
+                f"""conda init bash
         conda activate {self.condaenv}
         source ~/.bashrc"""
-            if self.condaenv
-            else (f"source {self.venv_path}" if self.venv_path else "")
+                if self.condaenv
+                else (f"source {self.venv_path}" if self.venv_path else "")
+            )
         )
 
         return (