Fix release

stack-info: PR: #51, branch: drisspg/stack/12

Author: drisspg

pyproject.toml

@@ -1,12 +1,12 @@
 # pyproject.toml
 
 [build-system]
-requires = ["hatchling"]
+requires = ["hatchling", "hatch-vcs"]
 build-backend = "hatchling.build"
 
 [project]
 name = "transformer_nuggets"
-version = "0.0.1"
+dynamic = ["version"]
 authors = [{ name = "Driss Guessous", email = "drisspguessous@gmail.com" }]
 description = "A place to store reusable transformer components found around the interwebs"
 readme = "README.md"
@@ -49,6 +49,8 @@ llama = [
 ]
 
 # ---------- TOOL CONFIGURATIONS ------------
+[tool.hatch.version]
+source = "vcs"
 
 # ---------- RUFF ------------
 [tool.ruff]

test/test_flash.py

@@ -1,6 +1,10 @@
+import importlib.util
 import pytest
 import torch
 
+if importlib.util.find_spec("triton") is None:
+    pytest.skip("Triton is not available", allow_module_level=True)
+
 from transformer_nuggets.flash import (
     attention,
     BiasMode,

transformer_nuggets/utils/benchmark.py

@@ -11,7 +11,6 @@
 
 from torch.cuda._memory_viz import profile_plot
 from torch.profiler import profile, ProfilerActivity, record_function, schedule
-from triton.testing import do_bench
 
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.NullHandler())