From 9bc0333ed5695b21a0ab1c6ded961615e187d2d4 Mon Sep 17 00:00:00 2001
From: Malo Bourgon <mbourgon@gmail.com>
Date: Thu, 20 Mar 2025 11:13:29 -0700
Subject: [PATCH] python312Packages.smolagents: disable test that requires
 missing dep `mlx-lm`

---
 .../python-modules/smolagents/default.nix     | 38 +++++++++++--------
 1 file changed, 22 insertions(+), 16 deletions(-)

diff --git a/pkgs/development/python-modules/smolagents/default.nix b/pkgs/development/python-modules/smolagents/default.nix
index 678f2e9fb46a..b7b4e50ae1c0 100644
--- a/pkgs/development/python-modules/smolagents/default.nix
+++ b/pkgs/development/python-modules/smolagents/default.nix
@@ -1,5 +1,6 @@
 {
   lib,
+  stdenv,
   accelerate,
   buildPythonPackage,
   docker,
@@ -103,22 +104,27 @@ buildPythonPackage rec {
 
   pythonImportsCheck = [ "smolagents" ];
 
-  disabledTests = [
-    # Missing dependencies
-    "test_ddgs_with_kwargs"
-    "test_e2b_executor_instantiation"
-    "test_flatten_messages_as_text_for_all_models"
-    "test_from_mcp"
-    "test_import_smolagents_without_extras"
-    "test_vision_web_browser_main"
-    # Tests require network access
-    "test_agent_type_output"
-    "test_can_import_sklearn_if_explicitly_authorized"
-    "test_transformers_message_no_tool"
-    "test_transformers_message_vl_no_tool"
-    "test_transformers_toolcalling_agent"
-    "test_visit_webpage"
-  ];
+  disabledTests =
+    [
+      # Missing dependencies
+      "test_ddgs_with_kwargs"
+      "test_e2b_executor_instantiation"
+      "test_flatten_messages_as_text_for_all_models"
+      "test_from_mcp"
+      "test_import_smolagents_without_extras"
+      "test_vision_web_browser_main"
+      # Tests require network access
+      "test_agent_type_output"
+      "test_can_import_sklearn_if_explicitly_authorized"
+      "test_transformers_message_no_tool"
+      "test_transformers_message_vl_no_tool"
+      "test_transformers_toolcalling_agent"
+      "test_visit_webpage"
+    ]
+    ++ lib.optionals stdenv.isDarwin [
+      # Missing dependencies
+      "test_get_mlx"
+    ];
 
   meta = {
     description = "Barebones library for agents";