chore(deps): remove unnecessary deps

Author: 34j

pyproject.toml

@@ -26,7 +26,6 @@ classifiers = [
 
 dependencies = [
     "array-api-compat>=1.11.2",
-    "cm-time>=0.1.2",
     "frozendict>=2.4.6",
 ]
 urls."Bug Tracker" = "https://github.com/34j/array-api-jit/issues"
@@ -36,6 +35,7 @@ urls.repository = "https://github.com/34j/array-api-jit"
 
 [dependency-groups]
 dev = [
+  "cm-time>=0.1.2",
   "jax>=0.4.30",
   "numba>=0.60.0",
   "pytest>=8,<9",

src/array_api_jit/_main.py

@@ -1,4 +1,3 @@
-# https://github.com/search?q=gumerov+translation+language%3APython&type=code&l=Python
 import importlib.util
 import warnings
 from collections.abc import Callable, Mapping, Sequence
@@ -14,7 +13,6 @@
     is_numpy_namespace,
     is_torch_namespace,
 )
-from frozendict import frozendict
 
 if importlib.util.find_spec("numba"):
     import numpy as np
@@ -123,8 +121,8 @@ def jit(
     """
 
     def new_decorator(f: Callable[Pinner, Tinner]) -> Callable[Pinner, Tinner]:
-        decorator_args_ = frozendict(decorator_args or {})
-        decorator_kwargs_ = frozendict(decorator_kwargs or {})
+        decorator_args_ = decorator_args or {}
+        decorator_kwargs_ = decorator_kwargs or {}
         decorator_ = decorator or {}
 
         @cache

tests/test_main.py

@@ -116,7 +116,7 @@ def test_jit(xp: Any) -> None:
         print(xp.__name__, name)
         for i in range(20):
             with timer() as timer_:
-                x = xp.arange(1000000, dtype=xp.float32)
+                x = xp.arange(1000000, dtype=xp.float32) / 1000000
                 p = func(x, 10)
                 assert p.shape == (1000000, 10)
             t[name] = timer_.elapsed