python3Packages.mlx-lm: 0.26.3 -> 0.30.2 (#479019)

pkgs/development/python-modules/mlx-lm/default.nix

@@ -21,22 +21,25 @@
   writableTmpDirAsHomeHook,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "mlx-lm";
-  version = "0.26.3";
+  version = "0.30.2";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "ml-explore";
     repo = "mlx-lm";
-    tag = "v${version}";
-    hash = "sha256-O4wW7wvIqSeBv01LoUCHm0/CgcRc5RfFHjvwyccp6UM=";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-6WlKAchze5B724XYwzpVHy+17HlMcGSYjJw0aOdm5yw=";
   };
 
   build-system = [
     setuptools
   ];
 
+  pythonRelaxDeps = [
+    "transformers"
+  ];
   dependencies = [
     jinja2
     mlx
@@ -53,9 +56,7 @@ buildPythonPackage rec {
     writableTmpDirAsHomeHook
   ];
 
-  pythonImportsCheck = [
-    "mlx_lm"
-  ];
+  pythonImportsCheck = [ "mlx_lm" ];
 
   disabledTestPaths = [
     # Requires network access to huggingface.co
@@ -65,22 +66,26 @@ buildPythonPackage rec {
     "tests/test_tokenizers.py"
     "tests/test_utils.py::TestUtils::test_convert"
     "tests/test_utils.py::TestUtils::test_load"
-    "tests/test_utils_load_model.py"
     "tests/test_prompt_cache.py::TestPromptCache::test_cache_to_quantized"
     "tests/test_prompt_cache.py::TestPromptCache::test_cache_with_generate"
     "tests/test_prompt_cache.py::TestPromptCache::test_trim_cache_with_generate"
+
     # RuntimeError: [metal_kernel] No GPU back-end.
     "tests/test_losses.py"
     "tests/test_models.py::TestModels::test_bitnet"
+
+    # TypeError: 'NoneType' object is not callable
+    "tests/test_models.py::TestModels::test_gated_delta"
+    "tests/test_models.py::TestModels::test_gated_delta_masked"
   ];
 
   meta = {
     description = "Run LLMs with MLX";
     homepage = "https://github.com/ml-explore/mlx-lm";
-    changelog = "https://github.com/ml-explore/mlx-lm/releases/tag/v${version}";
+    changelog = "https://github.com/ml-explore/mlx-lm/releases/tag/${finalAttrs.src.tag}";
     license = lib.licenses.mit;
     platforms = [
       "aarch64-darwin"
     ];
   };
-}
+})