Some fixes

Author: ASKabalan

pyproject.toml

@@ -39,15 +39,13 @@ dependencies = [
 ]
 dynamic = [
     "version",
-    "dependencies",
-    "optional-dependencies",
 ]
 keywords = [
 ]
 name = "s2fft"
 readme = ".pip_readme.rst"
 requires-python = ">=3.8"
-license.file = "LICENSE.txt"
+license.file = "LICENCE.txt"
 urls.homepage = "https://github.com/astro-informatics/s2fft"
 
 [project.optional-dependencies]
@@ -103,7 +101,6 @@ filterwarnings = [
 
 [tool.setuptools]
 packages = ["s2fft"]
-write_to = "s2fft/_version.py"
 
 [tool.setuptools.package-data]
 s2fft = ["default-logging-config.yaml"]

s2fft/utils/__init__.py

@@ -7,7 +7,7 @@
 from . import healpix_ffts
 from . import signal_generator
 from . import rotation
-from . import jax_pritimive
+from . import jax_primitive
 
 from jax.lib import xla_client
 from s2fft_lib import _s2fft

s2fft/utils/healpix_ffts.py

@@ -5,7 +5,7 @@
 import torch
 from s2fft.sampling import s2_samples as samples
 from functools import partial
-from s2fft.utils.jax_pritimive import BasePrimitive, register_primitive
+from s2fft.utils.jax_primitive import BasePrimitive, register_primitive
 import jaxlib.mlir.ir as ir
 from s2fft_lib import _s2fft
 from jaxlib.hlo_helpers import custom_call