more fixes

Author: JyotinderSingh

keras_remote/__init__.py

@@ -5,4 +5,4 @@
 os.environ.setdefault("GLOG_minloglevel", "3")
 os.environ.setdefault("GRPC_ENABLE_FORK_SUPPORT", "0")
 
-from keras_remote.core.core import run
+from keras_remote.core.core import run as run

keras_remote/backend/gke_client.py

@@ -68,21 +68,21 @@ def submit_k8s_job(
         f"Permission denied creating K8s Job. Ensure your kubeconfig "
         f"has 'create' permission for Jobs in namespace '{namespace}'. "
         f"Run: kubectl auth can-i create jobs -n {namespace}"
-      )
+      ) from e
     elif e.status == 404:
       raise RuntimeError(
         f"Namespace '{namespace}' not found. Create it with: "
         f"kubectl create namespace {namespace}"
-      )
+      ) from e
     elif e.status == 409:
       raise RuntimeError(
         f"Job '{job_name}' already exists. "
         f"Clean up with: kubectl delete job {job_name} -n {namespace}"
-      )
+      ) from e
     else:
       raise RuntimeError(
         f"Kubernetes API error: {e.status} - {e.reason}: {e.body}"
-      )
+      ) from e
 
 
 def wait_for_job(job, namespace="default", timeout=3600, poll_interval=10):
@@ -118,7 +118,7 @@ def wait_for_job(job, namespace="default", timeout=3600, poll_interval=10):
     try:
       job_status = batch_v1.read_namespaced_job_status(job_name, namespace)
     except ApiException as e:
-      raise RuntimeError(f"Failed to read job status: {e.reason}")
+      raise RuntimeError(f"Failed to read job status: {e.reason}") from e
 
     # Check completion conditions
     if job_status.status.succeeded and job_status.status.succeeded >= 1:
@@ -232,7 +232,7 @@ def _load_kube_config():
       f"Failed to load Kubernetes configuration. "
       f"Ensure you have run 'gcloud container clusters get-credentials <cluster-name>' "
       f"or have a valid kubeconfig. Error: {e}"
-    )
+    ) from e
 
 
 def _create_job_spec(

keras_remote/cli/commands/up.py

@@ -64,7 +64,7 @@ def up(project, zone, accelerator, cluster_name, yes):
     try:
       accel_config = accelerators.parse_accelerator(accelerator)
     except ValueError as e:
-      raise click.BadParameter(str(e), param_hint="--accelerator")
+      raise click.BadParameter(str(e), param_hint="--accelerator") from e
   else:
     accel_config = prompt_accelerator()
 

pyproject.toml

@@ -64,4 +64,7 @@ line-length = 80
 
 [tool.ruff.lint]
 select = ["B", "E", "F", "N", "PYI", "T20", "TID", "SIM", "W", "I", "NPY"]
-ignore = ["E501"]
+ignore = ["E501"]
+
+[tool.ruff.lint.per-file-ignores]
+"examples/*" = ["T201", "NPY002"]