move examples to root

lucidrains · Feb 5, 2025 · dbc0d95 · dbc0d95
1 parent 16a743c
commit dbc0d95
Show file tree

Hide file tree

Showing 10 changed files with 60 additions and 167 deletions.
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
@@ -25,7 +25,7 @@ jobs:
         python-version: '3.x'
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip uv
+        python -m pip install --upgrade pip
         pip install build
     - name: Build package
       run: python -m build

diff --git a/.github/workflows/python-test.yaml b/.github/workflows/python-test.yaml
@@ -27,7 +27,7 @@ jobs:
       run: |
         python -m pip install --upgrade pip uv
         python -m uv pip install torch==2.1.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu
-        python -m uv pip install .
+        python -m uv pip install .[test]
     - name: Test with pytest
       run: |
-        python setup.py test
+        pytest tests
diff --git a/examples/enwik8_simple/data/README.md b/examples/enwik8_simple/data/README.md
diff --git a/examples/enwik8_simple/data/enwik8.gz b/examples/enwik8_simple/data/enwik8.gz
diff --git a/examples/enwik8_simple/train_nar.py b/examples/enwik8_simple/train_nar.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,57 @@
+[project]
+name = "x-transformers"
+version = "2.0.0"
+description = "X-Transformers"
+authors = [
+    { name = "Phil Wang", email = "[email protected]" }
+]
+readme = "README.md"
+requires-python = ">= 3.9"
+license = { file = "LICENSE" }
+keywords = [
+    'artificial intelligence',
+    'attention mechanism',
+    'transformers'
+  ]
+classifiers=[
+    'Development Status :: 4 - Beta',
+    'Intended Audience :: Developers',
+    'Topic :: Scientific/Engineering :: Artificial Intelligence',
+    'License :: OSI Approved :: MIT License',
+    'Programming Language :: Python :: 3.6',
+  ]
+
+dependencies = [
+    'einx>=0.3.0',
+    'einops>=0.8.0',
+    'loguru',
+    'packaging>=21.0',
+    'torch>=2.0',
+]
+
+[project.urls]
+Homepage = "https://pypi.org/project/x-transformers/"
+Repository = "https://github.com/lucidrains/x-transformers"
+
+[project.optional-dependencies]
+examples = ["tqdm", "torchvision"]
+
+test = [
+    "pytest",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+
+[tool.pytest.ini_options]
+pythonpath = [
+  "."
+]
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.hatch.build.targets.wheel]
+packages = ["x_transformers"]
diff --git a/setup.cfg b/setup.cfg
diff --git a/setup.py b/setup.py
diff --git a/examples/toy_tasks/enc_dec_copy.py → train_copy.py b/examples/toy_tasks/enc_dec_copy.py → train_copy.py
diff --git a/examples/enwik8_simple/train.py → train_enwik8.py b/examples/enwik8_simple/train.py → train_enwik8.py