Update

Author: MTDickens

apply_configuration.py

@@ -33,9 +33,7 @@ def _replace_all_occurences(
     # Get files in this repository (e.g., exclude venv/).
     known_files: set[Path] = set()
     outer_dir = Path(".").parent.resolve()
-    proc = subprocess.run(
-        ["git", "ls-files"], encoding="utf-8", stdout=subprocess.PIPE, check=True
-    )
+    proc = subprocess.run(["git", "ls-files"], encoding="utf-8", stdout=subprocess.PIPE, check=True)
     for line in proc.stdout.split("\n"):
         known_files.add((outer_dir / line).resolve())
     known_files -= exclude
@@ -64,17 +62,11 @@ def _main() -> None:
     github_username = config["github-username"]
     assert " " not in github_username, "Malformed GitHub username"
     package_name = config["your-package-name"]
-    assert " " not in package_name, (
-        "Package names cannot contain spaces (you want to `import package_name`)"
-    )
-    assert "-" not in package_name, (
-        "Package names cannot dashes (you want to `import package_name`)"
-    )
+    assert " " not in package_name, "Package names cannot contain spaces (you want to `import package_name`)"
+    assert "-" not in package_name, "Package names cannot dashes (you want to `import package_name`)"
     python_version = config["python-version"]
     assert python_version.startswith("3"), "Only Python 3 is supported"
-    assert python_version.startswith("3."), (
-        "Missing dot in Python version (example: 3.10)"
-    )
+    assert python_version.startswith("3."), "Missing dot in Python version (example: 3.10)"
     python_subversion = python_version.split(".")[1]
     assert python_subversion.isdigit()
 
@@ -87,15 +79,9 @@ def _main() -> None:
         git_config_file = git_repo / "config"
         with open(git_config_file, encoding="utf-8") as fp:
             git_config_contents = fp.read()
-        if (
-            "git@github.com:MTDickens/research-code-python-starter-template"
-            in git_config_contents
-        ):
+        if "git@github.com:MTDickens/research-code-python-starter-template" in git_config_contents:
             shutil.rmtree(git_repo)
-        elif (
-            "https://github.com/MTDickens/research-code-python-starter-template"
-            in git_config_contents
-        ):
+        elif "https://github.com/MTDickens/research-code-python-starter-template" in git_config_contents:
             shutil.rmtree(git_repo)
 
     # Initialize the repo anew.

pyproject.toml

@@ -3,33 +3,28 @@ requires = ["setuptools", "setuptools-scm"]
 build-backend = "setuptools.build_meta"
 
 [project]
-name = "python_starter"
+name = "sym-img"
 version = "0.1.0"
-description = "A basic starter for Python packages."
+description = "Code for symbolic imagination paper"
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.12"
 dependencies = [
-   "jaxtyping>=0.3.7",
-   "matplotlib",
-   "numpy",
+    "hydra-core>=1.3.2",
+    "ipdb>=0.13.13",
 ]
 
 [project.optional-dependencies]
-develop = [
-    "pytest>=7.2.2",
-    "ruff",
-    "ty",
-]
+develop = ["pytest>=7.2.2", "pytest-xdist>=3.8.0", "ruff", "ty"]
 
 [tool.setuptools.packages.find]
 where = ["src"]
 
 [tool.setuptools.package-data]
-python_starter = ["py.typed"]
+sym_img = ["py.typed"]
 
 [tool.ruff]
-line-length = 88
-target-version = "py310"
+line-length = 128
+target-version = "py312"
 
 [tool.ruff.lint]
 select = ["E", "F", "W", "I", "D", "UP", "PL"]
@@ -39,8 +34,13 @@ ignore = [
     "F722", # jaxtyping uses strings in annotations that are not forward references (see ruff#9860)
 ]
 
+[tool.ruff.lint.pylint]
+max-args = 10
+max-branches = 30
+max-statements = 150
+
 [tool.ruff.lint.isort]
-known-first-party = ["python_starter"]
+known-first-party = ["sym_img"]
 
 [tool.ruff.lint.pydocstyle]
 convention = "numpy"
@@ -52,10 +52,16 @@ docstring-code-format = true
 docstring-code-line-length = "dynamic"
 
 [tool.ty.environment]
-python-version = "3.10"
+python-version = "3.12"
 
 [tool.ty.src]
 include = ["src", "tests"]
 
 [tool.ty.analysis]
 respect-type-ignore-comments = true
+
+[tool.uv.sources]
+sym-img = { workspace = true }
+
+[dependency-groups]
+dev = ["sym-img"]

run_ci_checks.sh

@@ -1,9 +1,11 @@
 #!/bin/bash                               # Run with bash (consistent with the rest of the repo)
 set -euo pipefail                         # Fail fast: -e stop on error, -u error on unset vars, pipefail propagate pipeline failures
 
+PYTEST_WORKERS="${PYTEST_WORKERS:-auto}"  # Allow local override, default to xdist auto worker count in CI checks
+
 uv sync --extra develop                   # Ensure the uv environment exists and all dev tools are installed
 
 uv run --no-sync ruff format --check .    # Check formatting without modifying files (CI-style formatting gate)
 uv run --no-sync ruff check .             # Run Ruff lint checks without autofixing (CI-style lint gate)
 uv run --no-sync ty check                 # Run ty static type checking
-uv run --no-sync pytest tests/            # Run unit tests in tests/ only
+uv run --no-sync pytest -n "${PYTEST_WORKERS}" --dist worksteal -s tests/  # Run unit tests in parallel by default