jazari 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

jazari/commands/init.py

@@ -1,3 +1,4 @@
+import os
 from rich.console import Console
 from rich.prompt import Prompt, Confirm
 from jazari.config_generator import load_config, make_config_dir, CONFIG_FILE
@@ -9,6 +10,10 @@ def init_command():
     Initialize Jazari config for this cluster.
     """
     current_config = load_config()
+    config_dir = os.path.expanduser('~/.jazari')
+    if not os.path.exists(config_dir):
+        os.makedirs(config_dir)
+        os.chmod(config_dir, 0o700)
 
     console.print('[bold blue]Welcome to Jazari setup.[/bold blue]')
     console.print(f"This will create a configuration file at [dim]{CONFIG_FILE}[/dim]\n")

jazari/commands/run.py

@@ -54,7 +54,6 @@ def run_command(
         time_limit      = time,
         job_name        = name,
         account_name    = account,
-        wandb_api_key   = wandb_key,
         hf_repo_id      = push_to_hub,
         hf_dataset_id   = pull_data,
         user_command    = full_command_str
@@ -68,6 +67,10 @@ def run_command(
     else:
         console.print("[dim]Submitting to Slurm scheduler...[/dim]")
 
+        submit_env = os.environ.copy()
+        if wandb_key:
+            submit_env['WANDB_API_KEY'] = wandb_key
+
         with tempfile.NamedTemporaryFile(mode='w+', suffix=".sh", delete=False) as temp_file:
             temp_script_path = temp_file.name
             temp_file.write(sbatch_content)
@@ -77,7 +80,8 @@ def run_command(
                 ["sbatch", temp_script_path],
                 check           = True,
                 capture_output  = True,
-                text            = True
+                text            = True,
+                env             = submit_env
             )
             job_id = result.stdout.strip().split()[-1]
             console.print(f"[bold green]✅ Job submitted successfully![/bold green] (ID: [bold]{job_id}[/bold])")

jazari/main.py

@@ -1,10 +1,12 @@
-import typer
+import typer, os
 from jazari.commands.status import status_command
 from jazari.commands.kill import kill_command
 from jazari.commands.logs import logs_command
 from jazari.commands.init import init_command
 from jazari.commands.run import run_command
 
+os.umask(0o077)
+
 # Initializing the Typer app
 app = typer.Typer(
     name            = "jazari",
@@ -13,7 +15,7 @@ app = typer.Typer(
     no_args_is_help = True
 )
 
-# Registering the commands
+# Register the commands
 app.command(name="status")(status_command)
 app.command(name="kill")(kill_command)
 app.command(name="logs")(logs_command)

jazari/slurm_generator.py

@@ -14,7 +14,6 @@ def generate_sbatch_script(
         time_limit:     str,
         job_name:       str,
         account_name:   Optional[str],
-        wandb_api_key:  Optional[str],
         hf_repo_id:     Optional[str],
         hf_dataset_id:  Optional[str],
         user_command:   str
@@ -34,7 +33,6 @@ def generate_sbatch_script(
         time_limit          = time_limit,
         job_name            = job_name,
         account_name        = account_name,
-        wandb_api_key       = wandb_api_key,
         hf_repo_id          = hf_repo_id,
         hf_dataset_id       = hf_dataset_id,
         user_command_script = user_command

{jazari-0.1.1.dist-info → jazari-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jazari
-Version: 0.1.1
+Version: 0.1.2
 Summary: The orchestration layer for modern Slurm clusters.
 Author-email: Levent Ozbek <levent@jazari.run>
 License: MIT

jazari-0.1.2.dist-info/RECORD

@@ -0,0 +1,16 @@
+jazari/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jazari/config_generator.py,sha256=ZLT48mCkSxoA5poURDhy4L7solmq5ZVXnBMkYDhpyDw,940
+jazari/main.py,sha256=prINKNdGeqh8GZzBgBoeC1d1pk2JIzYJL0OiDNckHC4,745
+jazari/slurm_generator.py,sha256=1powpTR6YbkUMb-eWacz2LweZRQIstB0G7Pv1N8lBtY,1401
+jazari/utils.py,sha256=kMgO01TSGUnQqwnpGznvX99L_pEl86rX602tvpSnZZA,911
+jazari/commands/init.py,sha256=0tpPE2Vzk76bKosTSsUOBq74LzgXVnT28ZaftWZPKXI,1488
+jazari/commands/kill.py,sha256=UsK4lGYTh5SrnaK87avFXgTyXrjf3ExiUZT6jkvA76I,1191
+jazari/commands/logs.py,sha256=RJEnmQ1CAzBj47w_TGtFYgB4WhgIKwlGP5kD3lPNnDI,1408
+jazari/commands/run.py,sha256=VWrX8-p5B4T18ZgSonZlZ3Uown8Nt-B5CKLGwiIf98o,4749
+jazari/commands/status.py,sha256=JODRwFkXkSHqqmZNZNDO93ZWHI2Kcrdrf7dbdOh3DnY,2252
+jazari-0.1.2.dist-info/licenses/LICENSE,sha256=OvKjjZV2nvR_v2QrhWR4TMgvtDmZYfZLFSt3bk1q6QA,1068
+jazari-0.1.2.dist-info/METADATA,sha256=bqAxIfy2pFP-OzA_kVRxS__3CCAa-4PmLNWeZgXvWx8,5826
+jazari-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+jazari-0.1.2.dist-info/entry_points.txt,sha256=9vYDaHn1_jBw_cMNC9ZKl4vRk3n2aDQ5dHL0cM8tEfE,43
+jazari-0.1.2.dist-info/top_level.txt,sha256=NY_Ke0lfKudUMc5w4rycAxzHKSgTUkhYqcmdFMYrya4,7
+jazari-0.1.2.dist-info/RECORD,,

jazari-0.1.1.dist-info/RECORD

@@ -1,16 +0,0 @@
-jazari/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jazari/config_generator.py,sha256=ZLT48mCkSxoA5poURDhy4L7solmq5ZVXnBMkYDhpyDw,940
-jazari/main.py,sha256=Qf8ORj5cX65fAEUgD2qbRnxqdnKdpLUAfVvcStaosRo,727
-jazari/slurm_generator.py,sha256=m5F76NqzMIwO4lL7MnyMWtuJTdj5Qv5qIN-P93CVKzM,1485
-jazari/utils.py,sha256=kMgO01TSGUnQqwnpGznvX99L_pEl86rX602tvpSnZZA,911
-jazari/commands/init.py,sha256=T8_kCUzH5BRnkDEqAzU49JjnBpvhimoDdU7ZYf5XCTo,1322
-jazari/commands/kill.py,sha256=UsK4lGYTh5SrnaK87avFXgTyXrjf3ExiUZT6jkvA76I,1191
-jazari/commands/logs.py,sha256=RJEnmQ1CAzBj47w_TGtFYgB4WhgIKwlGP5kD3lPNnDI,1408
-jazari/commands/run.py,sha256=gQqzjvCbanc11mBMce3MQ4doNsUO6KQOEptVLqKhWpI,4626
-jazari/commands/status.py,sha256=JODRwFkXkSHqqmZNZNDO93ZWHI2Kcrdrf7dbdOh3DnY,2252
-jazari-0.1.1.dist-info/licenses/LICENSE,sha256=OvKjjZV2nvR_v2QrhWR4TMgvtDmZYfZLFSt3bk1q6QA,1068
-jazari-0.1.1.dist-info/METADATA,sha256=twealsrdUYGqhl7_MI0zKjeiiiMPPBe0SerCa4AT8ZM,5826
-jazari-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-jazari-0.1.1.dist-info/entry_points.txt,sha256=9vYDaHn1_jBw_cMNC9ZKl4vRk3n2aDQ5dHL0cM8tEfE,43
-jazari-0.1.1.dist-info/top_level.txt,sha256=NY_Ke0lfKudUMc5w4rycAxzHKSgTUkhYqcmdFMYrya4,7
-jazari-0.1.1.dist-info/RECORD,,