slurmray 3.3.2__py3-none-any.whl → 3.5.0__py3-none-any.whl

slurmray/RayLauncher.py

@@ -132,10 +132,17 @@ class RayLauncher:
         # Determine the path to the file
         local_path = file_path
         local_path_from_pwd = os.path.relpath(local_path, self.pwd_path)
-        cluster_path = os.path.join("/users", self.server_username, "slurmray-server", ".slogs", "server", local_path_from_pwd)
+        cluster_path = os.path.join("/users", self.server_username, "slurmray-server", local_path_from_pwd)
         
         # Create the directory if not exists
-        ssh_client.exec_command(f"mkdir -p '{os.path.dirname(cluster_path)}'")
+        
+        stdin, stdout, stderr = ssh_client.exec_command(f"mkdir -p '{os.path.dirname(cluster_path)}'")
+        while True:
+            line = stdout.readline()
+            if not line:
+                break
+            print(line, end="")
+        time.sleep(1) # Wait for the directory to be created
         
         # Copy the file to the server
         sftp.put(file_path, cluster_path)
@@ -411,10 +418,11 @@ class RayLauncher:
             # Add slurmray --pre
             lines.append("slurmray --pre")
             # Adapt torch version
-            lines = [re.sub(r'\ntorch==.*', 'torch', line) for line in lines]
-            lines = [re.sub(r'\ntorchvision==.*', 'torchvision', line) for line in lines]
-            lines = [re.sub(r'\ntorchaudio==.*', 'torchaudio', line) for line in lines]
-            lines = [re.sub(r'\nbitsandbytes==.*', 'bitsandbytes', line) for line in lines]
+            lines = [re.sub(r'\ntorch==.*', '`\ntorch', line) for line in lines]
+            lines = [re.sub(r'\ntorchvision==.*', '\ntorchvision', line) for line in lines]
+            lines = [re.sub(r'\ntorchaudio==.*', '\ntorchaudio', line) for line in lines]
+            lines = [re.sub(r'\nbitsandbytes==.*', '\nbitsandbytes', line) for line in lines]
+            lines = [re.sub(r'\nslurmray==.*', '\n', line) for line in lines]
             
         with open(f"{self.project_path}/requirements.txt", 'w') as file:
             file.writelines(lines)

{slurmray-3.3.2.dist-info → slurmray-3.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: slurmray
-Version: 3.3.2
+Version: 3.5.0
 Summary: SlurmRay is a module for effortlessly distributing tasks on a Slurm cluster using the Ray library. 
 Home-page: https://henri-jamet.vercel.app/
 License: Apache License

{slurmray-3.3.2.dist-info → slurmray-3.5.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
-slurmray/RayLauncher.py,sha256=KqzDqAnKg_LmApYoqID_0bETMwFCEfPj_bQ-ErrvkM8,20251
+slurmray/RayLauncher.py,sha256=8qnPPB8zNybdT1fM0vK1BDX3xeUddO6IFXIPWbKBPwE,20548
 slurmray/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 slurmray/assets/sbatch_template.sh,sha256=c-7J4ItzrctDrbF5Znu8p1d_xIgayC9puhjX3nLMzsk,2273
 slurmray/assets/slurmray_server.sh,sha256=-Mt6CYjGdxui_wlD5NwKcv0wge23AjQqIscPbTA14mg,578
 slurmray/assets/slurmray_server_template.py,sha256=4VPJWq9sqW8BBarJB0XKvdQzbDomeHCIiqt0SBVnNIo,573
 slurmray/assets/spython_template.py,sha256=kRUvNQs9iCcg0wJLmm9LV0TnbUdlenZMYPr_bZPkXLg,597
-slurmray-3.3.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-slurmray-3.3.2.dist-info/METADATA,sha256=Aula9_eShrZcrtj_oNRkxPAl0_WyWnec3umnI7epAqk,2292
-slurmray-3.3.2.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-slurmray-3.3.2.dist-info/RECORD,,
+slurmray-3.5.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+slurmray-3.5.0.dist-info/METADATA,sha256=YoZpAEwgDBJrSDTjbIfashOuwFPCNvdC4F1Ft5CWCQY,2292
+slurmray-3.5.0.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
+slurmray-3.5.0.dist-info/RECORD,,