fix: test versioning

Author: flavioschneider

MANIFEST.in

@@ -0,0 +1 @@
+include aligner_pytorch/__init__.pxd

aligner_pytorch/__init__.pxd

@@ -1 +1 @@
-from .mas import mas 
+from aligner_pytorch.mas import mas

aligner_pytorch/__init__.py

@@ -1,17 +1,18 @@
-from .mas_c import compute_mas_alignement
-import torch 
-from torch import Tensor 
-from typing import Optional 
+from aligner_pytorch.mas_c import compute_mas_alignement
+import torch
+from torch import Tensor
+from typing import Optional
+
 
 def mas(x: Tensor, mask: Optional[Tensor] = None) -> Tensor:
-    b, m, n, device = *x.shape, x.device 
-    
-    values = x.clone().to(dtype=torch.float32, device='cpu').numpy() 
-    paths = torch.zeros_like(x, dtype=torch.int32, device='cpu').numpy() 
+    b, m, n, device = *x.shape, x.device
+
+    values = x.clone().to(dtype=torch.float32, device="cpu").numpy()
+    paths = torch.zeros_like(x, dtype=torch.int32, device="cpu").numpy()
 
     ms = torch.tensor([m], dtype=torch.int32).repeat(b).numpy()
     ns = torch.tensor([n], dtype=torch.int32).repeat(b).numpy()
 
     compute_mas_alignement(paths, values, ms, ns)
-    
+
     return torch.from_numpy(paths).to(device=device)

aligner_pytorch/mas.py

@@ -7,8 +7,8 @@
 
 setup(
     name="aligner-pytorch",
-    packages=find_packages(exclude=[]),
-    version="0.0.2",
+    version="0.0.14",
+    packages=find_packages(),
     license="MIT",
     description="Aligner - PyTorch",
     long_description=long_description,
@@ -18,14 +18,14 @@
     url="https://github.com/archinetai/audio-diffusion-pytorch",
     keywords=["artificial intelligence", "deep learning", "TTS", "alignment"],
     install_requires=[
-        "numpy",
         "torch>=1.6",
         "data-science-types>=0.2",
         "einops>=0.4",
         "einops-exts>=0.0.3",
     ],
-    include_dirs=["."],
-    ext_modules=cythonize(["aligner_pytorch/*.pyx"]),
+    setup_requires=["cython"],
+    include_dirs=["aligner_pytorch"],
+    ext_modules=cythonize(["aligner_pytorch/mas_c.pyx"]),
     classifiers=[
         "Development Status :: 4 - Beta",
         "Intended Audience :: Developers",