slurmthat 0.0.1__tar.gz

slurmthat-0.0.1/PKG-INFO

@@ -0,0 +1,4 @@
+Metadata-Version: 2.4
+Name: slurmthat
+Version: 0.0.1
+Requires-Dist: os

slurmthat-0.0.1/README.md

@@ -0,0 +1,48 @@
+# slurmthat
+#TJ in development
+
+Group 21, CodeAstro 2026
+
+This package creates and submits an sbatch .sh file on a user-inputed computer/computing system. 
+
+#Example of a bash script to launch a job
+
+#!/bin/bash
+
+#SBATCH --account=arcc
+
+#SBATCH --qos=debug
+
+#SBATCH --time=0-00:10:00
+
+#SBATCH --nodes=1
+
+#SBATCH --ntasks=24
+
+#SBATCH --cpus-per-task=4
+
+#SBATCH --mem-per-cpu=2G
+
+#SBATCH --mail-user=cowboyjoe@uwyo.edu
+
+#SBATCH --mail-type=BEGIN,END,FAIL
+
+#SBATCH --get-user-env
+
+export OMP_PROC_BIND=true
+
+export OMP_PLACES=threads
+
+export OMP_NUM_THREADS=32
+
+module load miniconda3 
+
+module load gcc/14.2.0
+
+srun echo "Start Job Process"
+
+srun hostname
+
+srun sleep 30
+srun --cpu-bind=cores check-hybrid.gnu.pm
+srun echo "End Job Process"

slurmthat-0.0.1/pyproject.toml

@@ -0,0 +1,11 @@
+[build-system]
+requires = ["setuptools >= 61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "slurmthat"
+version = "0.0.1"
+
+dependencies = [
+ "os",
+]

slurmthat-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

slurmthat-0.0.1/slurm.py

@@ -0,0 +1,70 @@
+import os
+
+class job:
+
+    def __init__(self, HPC=None, email=None, dir=None, job=None, partition=None,
+                 account=None):
+        self.HPC = HPC
+        self.account = account
+        self.email = email
+        self.dir = dir
+        self.job = job
+        self.partition = partition
+        if self.dir is None:
+          self.dir = os.getcwd()
+        self.filename = os.path.join(self.dir, self.job + ".sh")
+
+    def write(self, fail=False, start=False, finish=False, out_file=None, 
+              err_file=None, time=32, cpus=10, nodes=1, memory=48):
+      sh_filename = self.filename
+
+      with open(sh_filename, 'w') as f:
+          f.write('#!/bin/bash\n')
+          if self.account is not None:
+            f.write(f'#SBATCH --account={self.account}\n')
+          if self.email is not None:
+            f.write(f'#SBATCH --mail-user={self.email}\n')
+          f.write(f'#SBATCH --job-name={self.job}\n')
+          if self.partition is not None:
+            f.write(f'#SBATCH -p {self.partition}\n')
+
+          if fail and finish and start:
+            f.write('#SBATCH --mail-type=FAIL,END,BEGIN\n')
+          elif fail and start:
+            f.write('#SBATCH --mail-type=FAIL,BEGIN\n')
+          elif fail and finish:
+            f.write('#SBATCH --mail-type=FAIL,END\n')
+          elif start and finish:
+            f.write('#SBATCH --mail-type=BEGIN,END\n')
+          elif fail:
+            f.write('#SBATCH --mail-type=FAIL\n')
+          elif start:
+            f.write('#SBATCH --mail-type=BEGIN\n')
+          elif finish:
+            f.write('#SBATCH --mail-type=END\n')
+          if out_file is None:
+            out_file = f'{sh_filename.split('.sh')[0]}.out'
+          f.write(f'#SBATCH --output={out_file}\n')
+          if err_file is None:
+            err_file = f'{sh_filename.split('.sh')[0]}.err'
+          f.write(f'#SBATCH --error={err_file}\n')
+          f.write(f'''
+#SBATCH --time={time}:00:00
+#SBATCH --mem={memory}G
+#SBATCH --cpus-per-task={cpus}
+#SBATCH --nodes={nodes}\n
+''')
+          print(f'wrote .sh file to {sh_filename}')
+          return sh_filename
+    def add_line(self, line):
+      with open(self.filename, 'a') as f:
+          f.write(f'{line}\n')
+
+
+
+    def run(self, conda=None):
+      if conda is not None:
+        with open(self.filename, 'a') as f:
+          f.write('source ~/.bashrc\n')
+          f.write(f'conda activate {conda}\n')
+      os.system(f'sbatch {self.filename}')

slurmthat-0.0.1/slurmthat.egg-info/PKG-INFO

@@ -0,0 +1,4 @@
+Metadata-Version: 2.4
+Name: slurmthat
+Version: 0.0.1
+Requires-Dist: os

slurmthat-0.0.1/slurmthat.egg-info/SOURCES.txt

@@ -0,0 +1,8 @@
+README.md
+pyproject.toml
+slurm.py
+slurmthat.egg-info/PKG-INFO
+slurmthat.egg-info/SOURCES.txt
+slurmthat.egg-info/dependency_links.txt
+slurmthat.egg-info/requires.txt
+slurmthat.egg-info/top_level.txt

slurmthat-0.0.1/slurmthat.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

slurmthat-0.0.1/slurmthat.egg-info/requires.txt

@@ -0,0 +1 @@
+os

slurmthat-0.0.1/slurmthat.egg-info/top_level.txt

@@ -0,0 +1 @@
+slurm