haliax 1.4.dev366__tar.gz → 1.4.dev367__tar.gz

{haliax-1.4.dev366 → haliax-1.4.dev367}/AGENTS.md

@@ -25,7 +25,8 @@ repository. Follow these notes when implementing new features or fixing bugs.
 * **Formatting and Linting**: We use `ruff` via `pre-commit`.
 * **Typing**: the code base uses `mypy` for static type checking. `mypy` is run by pre‑commit and the
   configuration is found in `pyproject.toml`.
-* **Run `pre-commit run --all-files`** before committing. The CI workflows run the same checks.
+* **Run `uv run pre-commit run --all-files`** before committing. The CI workflows run the same checks.
+* **Use `uv run` for commands.** When running tools like `pytest` or other scripts, invoke them via `uv run` so the development dependencies are active.
 * **Doc Strings**: All public functions, classes, and modules should have docstrings, unless
   their purpose is painfully obvious. Use
   [Google style](https://google.github.io/styleguide/pyguide.html#38-comments-and-docstrings) for
@@ -44,12 +45,10 @@ repository. Follow these notes when implementing new features or fixing bugs.
 
 ## Testing
 
-* Tests are executed with `pytest`. The default workflow runs
-  `pytest tests -m "not entry and not slow and not ray"`.
+* Tests are executed with `pytest`. The default workflow runs `uv run pytest tests`.
 * In general, never relax tolerances in floating point tests unless specifically discussed with the
   team. Use `assert_allclose` with appropriate tolerances for numerical comparisons. We typically use
   1e-4 for more complex modules, and 1e-5 for simpler ones.
-* Tests should be reasonably fast. Mark long-running tests with @pytest.mark.slow so they are excluded from the default suite.
 * Always mark tests that depend on pytorch with `@skip_if_no_torch` to ensure they are skipped
   when PyTorch is not available. This is particularly important for tests that require PyTorch-specific
   functionality.
@@ -59,11 +58,6 @@ repository. Follow these notes when implementing new features or fixing bugs.
 
 * **Generic code**: many utilities are written with Python generics and dataclasses. Where possible,
   write reusable functions or classes that operate over TypeVars instead of hard coding concrete types.
-* **Configurations**: configuration files are dataclasses loaded via `draccus`. Keep configs
-  declarative and typed.
-* **Reproducibility**: Levanter aims for deterministic training where possible. Avoid sources of
-  nondeterminism unless explicitly required.
-* Prefer Stacked with fold or scan over writing custom loops, for better compile times and gradient checkpointing support
 
 ## Library conventions
 - Haliax revolves around `NamedArray` and explicit `Axis` objects. Prefer APIs that accept

{haliax-1.4.dev366 → haliax-1.4.dev367}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haliax
-Version: 1.4.dev366
+Version: 1.4.dev367
 Summary: Named Tensors for Legible Deep Learning in JAX
 Project-URL: Homepage, https://github.com/stanford-crfm/haliax
 Project-URL: Bug Tracker, https://github.com/stanford-crfm/haliax/issues/
@@ -19,20 +19,6 @@ Requires-Dist: equinox>=0.10.6
 Requires-Dist: jaxtyping>=0.2.20
 Requires-Dist: jmp>=0.0.4
 Requires-Dist: safetensors>=0.4.3
-Provides-Extra: dev
-Requires-Dist: chex>=0.1.86; extra == 'dev'
-Requires-Dist: mkdocs-include-markdown-plugin; extra == 'dev'
-Requires-Dist: mkdocs-literate-nav>=0.6.0; extra == 'dev'
-Requires-Dist: mkdocs-macros-plugin>=0.7.0; extra == 'dev'
-Requires-Dist: mkdocs-material>=7.3.3; extra == 'dev'
-Requires-Dist: mkdocs>=1.4.3; extra == 'dev'
-Requires-Dist: mkdocstrings-python>=1.1.2; extra == 'dev'
-Requires-Dist: mkdocstrings>=0.22.0; extra == 'dev'
-Requires-Dist: mypy>=0.910; extra == 'dev'
-Requires-Dist: pre-commit; extra == 'dev'
-Requires-Dist: pygments; extra == 'dev'
-Requires-Dist: pymdown-extensions; extra == 'dev'
-Requires-Dist: pytest>=7.4.0; extra == 'dev'
 Description-Content-Type: text/markdown
 
 <!--haliax-intro-start-->

{haliax-1.4.dev366 → haliax-1.4.dev367}/pyproject.toml

@@ -31,13 +31,20 @@ dependencies = [
 ]
 dynamic =[ "version" ]
 
-[project.optional-dependencies]
-dev=["pytest >= 7.4.0", "mypy >= 0.910", "mkdocs >= 1.4.3", "mkdocs-material >= 7.3.3", "mkdocstrings >= 0.22.0",
-    "mkdocs-literate-nav >= 0.6.0", "mkdocs-macros-plugin >= 0.7.0", "mkdocstrings-python >= 1.1.2",
+
+[dependency-groups]
+dev = [
+    "pytest >= 7.4.0",
+    "mypy >= 0.910",
+    "mkdocs >= 1.4.3",
+    "mkdocs-material >= 7.3.3",
+    "mkdocstrings >= 0.22.0",
+    "mkdocs-literate-nav >= 0.6.0",
+    "mkdocs-macros-plugin >= 0.7.0",
+    "mkdocstrings-python >= 1.1.2",
     "mkdocs-include-markdown-plugin",
     "pymdown-extensions",
     "pygments",
-    "pymdown-extensions",
     "chex>=0.1.86",
     "pre-commit",
 ]

haliax-1.4.dev367/src/haliax/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.4.dev367"

haliax-1.4.dev366/src/haliax/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.4.dev366"