q8s 0.8.0.dev2__tar.gz → 0.9.0__tar.gz

{q8s-0.8.0.dev2 → q8s-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: q8s
-Version: 0.8.0.dev2
+Version: 0.9.0
 Summary: Kernel extension for executing quantum programs in simulators on q8s clusters
 Author-email: Vlad Stirbu <vstirbu@gmail.com>
 License:                                  Apache License
@@ -211,6 +211,11 @@ Keywords: jupyter,quantum,kernel
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
@@ -224,6 +229,7 @@ Requires-Dist: pluggy==1.5.0
 Requires-Dist: python-dotenv==1.0.1
 Requires-Dist: pyyaml==6.0.2
 Requires-Dist: typer>=0.15.4
+Requires-Dist: GitPython>=3.1.43
 Provides-Extra: benchmark
 Requires-Dist: pandas; extra == "benchmark"
 Requires-Dist: testbook; extra == "benchmark"

{q8s-0.8.0.dev2 → q8s-0.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "q8s"
-version = "0.8.0.dev2"
+version = "0.9.0"
 description = "Kernel extension for executing quantum programs in simulators on q8s clusters"
 authors = [{ name = "Vlad Stirbu", email = "vstirbu@gmail.com" }]
 readme = "README.md"
@@ -10,6 +10,11 @@ classifiers = [
   "Development Status :: 4 - Beta",
   "Intended Audience :: Developers",
   "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
   "License :: OSI Approved :: MIT License",
   "Operating System :: OS Independent",
 ]
@@ -22,6 +27,7 @@ dependencies = [
   "python-dotenv==1.0.1",
   "pyyaml==6.0.2",
   "typer>=0.15.4",
+  "GitPython>=3.1.43",
 ]
 requires-python = ">= 3.10"
 

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s/cli.py

@@ -16,6 +16,22 @@ from q8s.workload import Workload
 app = typer.Typer()
 
 
+@app.command()
+def init(
+    images: Annotated[
+        bool, typer.Option(help="Initialize images cache if build in a CI pipeline")
+    ] = False,
+):
+    project = Project()
+    project.init_cache()
+
+    if images:
+        project.images_from_ci()
+        project.update_images_cache()
+
+    print(f"Project {project.name} initialized")
+
+
 @app.command()
 def build(
     init: Annotated[bool, typer.Option(help="Initialize project")] = False,

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s/execution.py

@@ -146,38 +146,6 @@ class K8sContext:
         self.__progress.advance(prepare_task, 1)
         return job
 
-    def __create_config_map_object(self, job: client.V1Job, code: str):
-        """
-        Create a ConfigMap object with the given code.
-        """
-
-        data = {}
-        data["main.py"] = code
-
-        # Configureate ConfigMap from a local file
-        configmap = client.V1ConfigMap(
-            api_version="v1",
-            kind="ConfigMap",
-            data=data,
-            metadata=client.V1ObjectMeta(
-                name=self.name,
-                owner_references=[
-                    client.V1OwnerReference(
-                        api_version="batch/v1",
-                        kind="Job",
-                        name=job.metadata.name,
-                        uid=job.metadata.uid,
-                        # block_owner_deletion=True,
-                        # controller=True,
-                    )
-                ],
-            ),
-        )
-
-        self.core_api_instance.create_namespaced_config_map(
-            namespace=self.namespace, body=configmap
-        )
-
     def __create_config_map_object_from_workload(
         self, job: client.V1Job, workload: Workload
     ):

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s/plugins/cpu_job.py

@@ -1,4 +1,5 @@
 from typing import Dict
+from q8s.plugins.job import JobPlugin
 from q8s.plugins.job_template_spec import hookimpl
 from q8s.enums import Target
 from kubernetes import client
@@ -6,7 +7,7 @@ from q8s.constants import WORKSPACE
 from q8s.workload import Workload
 
 
-class CPUJobTemplatePlugin:
+class CPUJobTemplatePlugin(JobPlugin):
 
     @hookimpl
     def makejob(
@@ -25,13 +26,16 @@ class CPUJobTemplatePlugin:
 
         volume_name = f"app-volume-{name}"
 
+        env_var = list(env)
         if workload.is_src_project:
-            env.append(client.V1EnvVar(name="PYTHONPATH", value=f"{WORKSPACE}/src"))
+            env_var.append(client.V1EnvVar(name="PYTHONPATH", value=f"{WORKSPACE}/src"))
+
+        self.patch_environment_with_git_info(env_var)
 
         container = client.V1Container(
             name="quantum-routine",
             image=container_image,
-            env=env,
+            env=env_var,
             command=["python"],
             args=(
                 ["-m", workload.entry_module] + workload.args

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s/plugins/cuda_job.py

@@ -3,13 +3,14 @@ from typing import Dict
 from kubernetes import client
 from q8s.constants import WORKSPACE
 from q8s.enums import Target
+from q8s.plugins.job import JobPlugin
 from q8s.plugins.job_template_spec import hookimpl
 from q8s.workload import Workload
 
 MEMORY = os.environ.get("MEMORY", "32Gi")
 
 
-class CUDAJobTemplatePlugin:
+class CUDAJobTemplatePlugin(JobPlugin):
     """
     This plugin is used to create a job template for a GPU job.
     """
@@ -31,13 +32,16 @@ class CUDAJobTemplatePlugin:
 
         volume_name = f"app-volume-{name}"
 
+        env_var = list(env)
         if workload.is_src_project:
-            env.append(client.V1EnvVar(name="PYTHONPATH", value=f"{WORKSPACE}/src"))
+            env_var.append(client.V1EnvVar(name="PYTHONPATH", value=f"{WORKSPACE}/src"))
+
+        self.patch_environment_with_git_info(env_var)
 
         container = client.V1Container(
             name="quantum-routine",
             image=container_image,
-            env=env,
+            env=env_var,
             command=["python"],
             args=(
                 ["-m", workload.entry_module] + workload.args

q8s-0.9.0/q8s/plugins/job.py

@@ -0,0 +1,53 @@
+import os
+from kubernetes import client
+
+from q8s.plugins.utils.git_info import get_git_info
+
+
+class JobPlugin:
+    """
+    Base class for job template plugins.
+    This class provides functionality to patch job environment variables with Git information,
+    such as commit hash, branch name, and remote repository URL. It is intended to be extended
+    by specific job plugins that require Git metadata in their execution environment.
+    The following environment variables are added:
+        - Q8S_GIT_COMMIT, MLFLOW_GIT_COMMIT: The current Git commit hash.
+        - Q8S_GIT_BRANCH, MLFLOW_GIT_BRANCH: The current Git branch name.
+        - Q8S_GIT_REPO_URL, MLFLOW_GIT_REPO_URL: The remote repository URL.
+        - GIT_PYTHON_REFRESH: Set to "quiet" to suppress GitPython refresh warnings.
+    """
+
+    def patch_environment_with_git_info(
+        self, env: list[client.V1EnvVar]
+    ) -> list[client.V1EnvVar]:
+        """
+        Patch environment variables for the job with Git information
+
+        Args:
+            env (list[client.V1EnvVar]): Original environment variables
+        Returns:
+            list[client.V1EnvVar]: Patched environment variables
+        """
+
+        # Get Git information from the q8sctl current working directory
+        git_info = get_git_info(os.getcwd())
+
+        if git_info.commit:
+            env.append(client.V1EnvVar(name="Q8S_GIT_COMMIT", value=git_info.commit))
+            env.append(client.V1EnvVar(name="MLFLOW_GIT_COMMIT", value=git_info.commit))
+
+        if git_info.branch:
+            env.append(client.V1EnvVar(name="Q8S_GIT_BRANCH", value=git_info.branch))
+            env.append(client.V1EnvVar(name="MLFLOW_GIT_BRANCH", value=git_info.branch))
+
+        if git_info.remote_url:
+            env.append(
+                client.V1EnvVar(name="Q8S_GIT_REPO_URL", value=git_info.remote_url)
+            )
+            env.append(
+                client.V1EnvVar(name="MLFLOW_GIT_REPO_URL", value=git_info.remote_url)
+            )
+
+        env.append(client.V1EnvVar(name="GIT_PYTHON_REFRESH", value="quiet"))
+
+        return env

q8s-0.9.0/q8s/plugins/utils/git_info.py

@@ -0,0 +1,111 @@
+from dataclasses import dataclass
+from typing import Optional, Dict, Any
+import os
+
+from git import Repo, InvalidGitRepositoryError, NoSuchPathError
+
+
+@dataclass
+class GitExtraInfo:
+    """Additional Git information dataclass"""
+
+    branch_source: Optional[str]
+    git_dir: Optional[str]
+    working_tree_dir: Optional[str]
+
+
+@dataclass
+class GitExtraReason:
+    """Reason for missing Git information"""
+
+    reason: str
+
+
+@dataclass
+class GitInfo:
+    """Git information dataclass"""
+
+    commit: Optional[str]
+    branch: Optional[str]
+    remote_url: Optional[str]
+    extra: GitExtraInfo | GitExtraReason
+
+
+def _detect_branch_from_env() -> Optional[str]:
+    """
+    Try to infer branch name from common CI environment variables.
+    Extend this for your CI setup as needed.
+    """
+    # GitHub Actions
+    if ref_name := os.getenv("GITHUB_REF_NAME"):
+        return ref_name
+
+    # GitLab CI
+    if ref_name := os.getenv("CI_COMMIT_REF_NAME"):
+        return ref_name
+
+    # Generic
+    return os.getenv("BRANCH_NAME")
+
+
+def get_git_info(path: str = ".") -> GitInfo:
+    """
+    Collect commit, branch and remote URL info using GitPython.
+
+    Returns GitInfo with None fields if not in a git repo.
+    """
+    try:
+        repo = Repo(path, search_parent_directories=True)
+    except (InvalidGitRepositoryError, NoSuchPathError):
+        return GitInfo(
+            commit=None,
+            branch=None,
+            remote_url=None,
+            extra=GitExtraReason(reason="not_a_git_repo"),
+        )
+
+    # Commit SHA
+    try:
+        commit = repo.head.commit.hexsha
+    except ValueError:
+        commit = None
+
+    # Branch (handle detached HEAD, CI environments, etc.)
+    branch = None
+    if repo.head.is_detached:
+        branch = _detect_branch_from_env()
+        branch_source = "env" if branch else "detached_head"
+    else:
+        branch = repo.active_branch.name
+        branch_source = "repo"
+
+    # Remote URL
+    try:
+        origin = repo.remotes.origin
+        remote_url = origin.url
+        if remote_url.startswith("git@"):
+            # git@github.com:org/repo.git -> https://github.com/org/repo.git
+            host, _, url_path = remote_url.partition(":")
+            remote_url = f"https://{host.removeprefix('git@')}/{url_path}"
+        elif remote_url.startswith("ssh://"):
+            # ssh://git@github.com/org/repo.git -> https://github.com/org/repo.git
+            stripped = remote_url.removeprefix("ssh://")
+            if stripped.startswith("git@"):
+                stripped = stripped.removeprefix("git@")
+            host, _, url_path = stripped.partition("/")
+            remote_url = f"https://{host}/{url_path}"
+    except (AttributeError, IndexError, ValueError):
+        remote_url = None
+
+    extra = GitExtraInfo(
+        branch_source=branch_source,
+        git_dir=repo.git_dir,
+        working_tree_dir=repo.working_tree_dir,
+    )
+
+    return GitInfo(
+        commit=commit,
+        branch=branch,
+        remote_url=remote_url,
+        extra=extra,
+    )

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s/project.py

@@ -1,3 +1,4 @@
+import configparser
 from dataclasses import dataclass
 from datetime import datetime
 from pathlib import Path
@@ -43,7 +44,7 @@ def _read_stream_lines(stream, progress, silent, is_error: bool = False):
     """
     while True:
         line = stream.readline()
-        if line == '':
+        if line == "":
             break
         if not silent:
             if is_error:
@@ -96,6 +97,7 @@ class Q8SDocker:
     username: str
     registry: Optional[str]
 
+
 @dataclass
 class Q8SProject:
     name: str
@@ -112,9 +114,11 @@ class CacheNotBuiltException(Exception):
 class ProjectNotFoundException(Exception):
     pass
 
+
 class ProjectInvalidConfigurationException(Exception):
     pass
 
+
 class Project:
     name: str
     __path: str
@@ -215,14 +219,15 @@ class Project:
             stderr=STDOUT,
             bufsize=1,
             universal_newlines=True,
-            encoding="utf-8",   # force UTF-8 decoding
-            errors="replace",   # avoid crashing on bad bytes
+            encoding="utf-8",  # force UTF-8 decoding
+            errors="replace",  # avoid crashing on bad bytes
         )
 
         if sys.platform == "win32":
             # Windows-safe: read line by line instead of using selectors
             _handle_subprocess_output(build_process, progress, silent)
         else:
+
             def handle_output(stream, mask):
                 # Because the process' output is line buffered, there's only ever one
                 # line to read when this function is called
@@ -270,8 +275,8 @@ class Project:
             stderr=STDOUT,
             bufsize=1,
             universal_newlines=True,
-            encoding="utf-8",   # force UTF-8 decoding
-            errors="replace",   # avoid crashing on bad bytes
+            encoding="utf-8",  # force UTF-8 decoding
+            errors="replace",  # avoid crashing on bad bytes
         )
 
         if sys.platform == "win32":
@@ -279,6 +284,7 @@ class Project:
             _handle_subprocess_output(push_process, progress, silent)
 
         else:
+
             def handle_output(stream, mask):
                 # Because the process' output is line buffered, there's only ever one
                 # line to read when this function is called
@@ -306,11 +312,19 @@ class Project:
         else:
             progress.advance(task)
 
+    def images_from_ci(self):
+        """
+        Images are build in a CI pipeline.
+        """
+        for target in self.configuration.targets.keys():
+            self.__images[target] = self.__image_name(target)
+
     def update_images_cache(self):
         """
         Update the images cache
         """
         with open(join(self.__path, ".q8s_cache", "images"), "w") as f:
+            print(f"# This file is autogenerated by q8sctl\n", file=f)
             yaml.dump(self.__images, f)
 
     def clear_cache(self):
@@ -325,7 +339,7 @@ class Project:
 
     def __image_name(self, target: str):
         registry = self.configuration.docker.registry
-        username = self.__docker_login()
+        username = self.__docker_login().lower()
 
         # Build image name based on available information
         if registry and username:
@@ -388,6 +402,15 @@ class Project:
             file=f,
         )
         print(f"LABEL org.opencontainers.image.title={self.name}", file=f)
+
+        url = self.__get_project_url()
+
+        if url is not None:
+            print(
+                f"LABEL org.opencontainers.image.source={url.strip()}",
+                file=f,
+            )
+
         print("", file=f)
 
         print(f"WORKDIR {WORKSPACE}", file=f)
@@ -399,3 +422,20 @@ class Project:
             raise Exception(f"Target {target} not found")
 
         return getattr(self.configuration.targets, target)
+
+    def __get_project_url(self) -> Optional[str]:
+        """
+        Get the project URL from setup.cfg if available
+        """
+        cfg_path = Path(self.__path) / "setup.cfg"
+        if not cfg_path.exists():
+            return None
+
+        config = configparser.ConfigParser()
+        config.read(cfg_path)
+
+        url = config.get("metadata", "url", fallback=None)
+        if url is not None:
+            return url.strip()
+
+        return None

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s/utils.py

@@ -4,6 +4,11 @@ from q8s.project import CacheNotBuiltException, Project, ProjectNotFoundExceptio
 
 
 def extract_non_none_value(arr):
+    """
+    Extract the first non-None value from a list
+
+    Used to get the first available configuration value from multiple sources generated by job plugins
+    """
     non_none_values = [x for x in arr if x is not None]
     return non_none_values[0] if non_none_values else None
 

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s/workload.py

@@ -7,34 +7,34 @@ from .multifiles import collect_imported_files
 
 class Workload:
     """
-    Represents a mutifile workload consisting of an entry script and its imported files
+    Represents a multifile workload consisting of an entry script and its imported files
     """
 
-    __pyproject_root: Path | None = None
-    __args: list[str] = []
-
     def __init__(self, entry_script: Path | str = None, code: str = None):
+        self.__pyproject_root: Path | None = None
+        self.__args: list[str] = []
+
         if entry_script is not None:
 
-            if self.__is_src_layout():
-                self.__entry_script = Path(os.path.abspath(entry_script))
+            self.__entry_script = Path(os.path.abspath(entry_script))
+
+            if self.__detect_and_set_src_layout():
                 self.__base_path = self.__pyproject_root
 
                 self.__files = collect_imported_files(
                     self.__entry_script, is_project=True
                 )
-
             else:
-                self.__entry_script = Path(os.path.abspath(entry_script))
                 self.__base_path = self.__entry_script.parent
 
                 self.__files = collect_imported_files(
                     self.__entry_script, is_project=False
                 )
 
-            self.__data = {
-                self.__path_mapping(f): open(f, "r").read() for f in self.__files
-            }
+            self.__data = {}
+            for f in self.__files:
+                with open(f, "r", encoding="utf-8") as file:
+                    self.__data[self.__path_mapping(f)] = file.read()
 
         elif code is not None:
             self.__entry_script = Path(os.path.abspath("main.py"))
@@ -86,7 +86,7 @@ class Workload:
         """
         Entry script module name
         """
-        return self.entry_script.replace("/", ".")[:-3]  # strip .py
+        return self.entry_script.replace(os.path.sep, ".")[:-3]  # strip .py
 
     @property
     def data(self) -> dict[str, str]:
@@ -127,7 +127,7 @@ class Workload:
         """
         return self.__relative_path(file).replace("/", "__")
 
-    def __pyproject_root(self, start: Path | None = None) -> Path | None:
+    def __find_pyproject_root(self) -> Path | None:
         """
         Find the project root by looking for setup.cfg or pyproject.toml
         """
@@ -140,11 +140,11 @@ class Workload:
 
         return None
 
-    def __is_src_layout(self) -> bool:
+    def __detect_and_set_src_layout(self) -> bool:
         """
         Determine if the project uses a src/ layout
         """
-        root = self.__pyproject_root()
+        root = self.__find_pyproject_root()
 
         if root is None:
             return False

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: q8s
-Version: 0.8.0.dev2
+Version: 0.9.0
 Summary: Kernel extension for executing quantum programs in simulators on q8s clusters
 Author-email: Vlad Stirbu <vstirbu@gmail.com>
 License:                                  Apache License
@@ -211,6 +211,11 @@ Keywords: jupyter,quantum,kernel
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
@@ -224,6 +229,7 @@ Requires-Dist: pluggy==1.5.0
 Requires-Dist: python-dotenv==1.0.1
 Requires-Dist: pyyaml==6.0.2
 Requires-Dist: typer>=0.15.4
+Requires-Dist: GitPython>=3.1.43
 Provides-Extra: benchmark
 Requires-Dist: pandas; extra == "benchmark"
 Requires-Dist: testbook; extra == "benchmark"

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s.egg-info/SOURCES.txt

@@ -26,10 +26,15 @@ q8s/matplotlib/backend.py
 q8s/plugins/__init__.py
 q8s/plugins/cpu_job.py
 q8s/plugins/cuda_job.py
+q8s/plugins/job.py
 q8s/plugins/job_template_spec.py
+q8s/plugins/utils/__init__.py
+q8s/plugins/utils/git_info.py
 tests/__init__.py
 tests/test_backend.py
 tests/test_deps.py
+tests/test_git_info.py
+tests/test_job_plugin.py
 tests/test_job_template_spec.py
 tests/test_kernel.py
 tests/test_multifiles.py

{q8s-0.8.0.dev2 → q8s-0.9.0}/q8s.egg-info/requires.txt

@@ -6,6 +6,7 @@ pluggy==1.5.0
 python-dotenv==1.0.1
 pyyaml==6.0.2
 typer>=0.15.4
+GitPython>=3.1.43
 
 [benchmark]
 pandas

q8s-0.9.0/tests/test_git_info.py

@@ -0,0 +1,102 @@
+import os
+import tempfile
+import unittest
+from pathlib import Path
+
+from git import Actor, Repo
+
+from q8s.plugins.utils.git_info import (
+    GitExtraInfo,
+    GitExtraReason,
+    _detect_branch_from_env,
+    get_git_info,
+)
+
+
+class TestDetectBranchFromEnv(unittest.TestCase):
+
+    def test_prefers_github_ref_name(self):
+        with unittest.mock.patch.dict(
+            os.environ,
+            {
+                "GITHUB_REF_NAME": "github-branch",
+                "CI_COMMIT_REF_NAME": "gitlab-branch",
+                "BRANCH_NAME": "generic-branch",
+            },
+            clear=True,
+        ):
+            self.assertEqual(_detect_branch_from_env(), "github-branch")
+
+    def test_uses_gitlab_commit_ref_when_github_missing(self):
+        with unittest.mock.patch.dict(
+            os.environ,
+            {"CI_COMMIT_REF_NAME": "gitlab-branch", "BRANCH_NAME": "generic-branch"},
+            clear=True,
+        ):
+            self.assertEqual(_detect_branch_from_env(), "gitlab-branch")
+
+    def test_falls_back_to_generic_branch_name(self):
+        with unittest.mock.patch.dict(
+            os.environ,
+            {"BRANCH_NAME": "generic-branch"},
+            clear=True,
+        ):
+            self.assertEqual(_detect_branch_from_env(), "generic-branch")
+
+    def test_returns_none_when_no_env_vars_set(self):
+        with unittest.mock.patch.dict(os.environ, {}, clear=True):
+            self.assertIsNone(_detect_branch_from_env())
+
+
+class TestGetGitInfo(unittest.TestCase):
+
+    def _create_repo_with_commit(self):
+        tempdir = tempfile.TemporaryDirectory()
+        self.addCleanup(tempdir.cleanup)
+        repo = Repo.init(tempdir.name)
+        file_path = Path(tempdir.name) / "tracked.txt"
+        file_path.write_text("data\n", encoding="utf-8")
+        repo.index.add([str(file_path)])
+        actor = Actor("Tester", "tester@example.com")
+        repo.index.commit("initial", author=actor, committer=actor)
+        return tempdir.name, repo
+
+    def test_returns_reason_when_path_not_repo(self):
+        with tempfile.TemporaryDirectory() as tmpdir:
+            info = get_git_info(tmpdir)
+
+        self.assertIsNone(info.commit)
+        self.assertIsNone(info.branch)
+        self.assertIsNone(info.remote_url)
+        self.assertEqual(type(info.extra), GitExtraReason)
+        self.assertEqual(info.extra.reason, "not_a_git_repo")
+
+    def test_collects_repo_branch_commit_and_remote(self):
+        path, repo = self._create_repo_with_commit()
+        repo.create_remote("origin", "git@github.com:org/project.git")
+
+        info = get_git_info(path)
+
+        self.assertEqual(info.commit, repo.head.commit.hexsha)
+        self.assertEqual(info.branch, repo.active_branch.name)
+        self.assertEqual(info.remote_url, "https://github.com/org/project.git")
+        self.assertEqual(type(info.extra), GitExtraInfo)
+        self.assertEqual(info.extra.branch_source, "repo")
+        self.assertEqual(info.extra.git_dir, repo.git_dir)
+        self.assertEqual(info.extra.working_tree_dir, repo.working_tree_dir)
+
+    def test_detached_head_uses_env_branch(self):
+        path, repo = self._create_repo_with_commit()
+        repo.git.checkout(repo.head.commit.hexsha)  # detach HEAD
+
+        with unittest.mock.patch.dict(
+            os.environ, {"GITHUB_REF_NAME": "ci-branch"}, clear=True
+        ):
+            info = get_git_info(path)
+
+        self.assertEqual(info.branch, "ci-branch")
+        self.assertEqual(info.extra.branch_source, "env")
+
+
+if __name__ == "__main__":
+    unittest.main()

q8s-0.9.0/tests/test_job_plugin.py

@@ -0,0 +1,64 @@
+import unittest
+from unittest.mock import patch
+
+from kubernetes import client
+
+from q8s.plugins.job import JobPlugin
+from q8s.plugins.utils.git_info import GitInfo
+
+
+class TestJobPlugin(unittest.TestCase):
+
+    @patch("q8s.plugins.job.get_git_info")
+    @patch("q8s.plugins.job.os.getcwd")
+    def test_patch_environment_appends_git_metadata(
+        self,
+        mock_getcwd,
+        mock_get_git_info,
+    ):
+        mock_getcwd.return_value = "/tmp/repo"
+        mock_get_git_info.return_value = GitInfo(
+            commit="abc123",
+            branch="main",
+            remote_url="https://example.com/repo.git",
+            extra={},
+        )
+        env = [client.V1EnvVar(name="EXISTING", value="1")]
+
+        plugin = JobPlugin()
+        result = plugin.patch_environment_with_git_info(env)
+
+        self.assertIs(result, env)
+        mock_get_git_info.assert_called_once_with("/tmp/repo")
+        env_pairs = {(var.name, var.value) for var in env}
+        self.assertIn(("Q8S_GIT_COMMIT", "abc123"), env_pairs)
+        self.assertIn(("MLFLOW_GIT_COMMIT", "abc123"), env_pairs)
+        self.assertIn(("Q8S_GIT_BRANCH", "main"), env_pairs)
+        self.assertIn(("MLFLOW_GIT_BRANCH", "main"), env_pairs)
+        self.assertIn(("Q8S_GIT_REPO_URL", "https://example.com/repo.git"), env_pairs)
+        self.assertIn(
+            ("MLFLOW_GIT_REPO_URL", "https://example.com/repo.git"), env_pairs
+        )
+        self.assertIn(("GIT_PYTHON_REFRESH", "quiet"), env_pairs)
+
+    @patch("q8s.plugins.job.get_git_info")
+    def test_patch_environment_handles_missing_git_metadata(self, mock_get_git_info):
+        mock_get_git_info.return_value = GitInfo(
+            commit=None,
+            branch=None,
+            remote_url=None,
+            extra={"reason": "Not a git repository"},
+        )
+        env: list[client.V1EnvVar] = []
+
+        plugin = JobPlugin()
+        result = plugin.patch_environment_with_git_info(env)
+
+        self.assertIs(result, env)
+        self.assertEqual(
+            [(var.name, var.value) for var in env], [("GIT_PYTHON_REFRESH", "quiet")]
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()

{q8s-0.8.0.dev2 → q8s-0.9.0}/tests/test_job_template_spec.py

@@ -1,8 +1,11 @@
 import unittest
 from unittest.mock import patch, MagicMock
 from kubernetes import client
-from q8s.plugins.job_template_spec import CPUJobTemplatePlugin, CUDAJobTemplatePlugin
+
+# from q8s.plugins.job_template_spec import CPUJobTemplatePlugin, CUDAJobTemplatePlugin
 from q8s.enums import Target
+from q8s.plugins.cpu_job import CPUJobTemplatePlugin
+from q8s.plugins.cuda_job import CUDAJobTemplatePlugin
 
 
 class TestCPUandGPUJobTemplatePlugin(unittest.TestCase):