From fe8292433a65dc7fffd7ad745ba10eda4eeea0f5 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 21 Sep 2025 21:35:54 +0000
Subject: [PATCH] python3Packages.open-clip-torch: 3.1.0 -> 3.2.0

Diff: https://github.com/mlfoundations/open_clip/compare/v3.1.0...v3.2.0

Changelog: https://github.com/mlfoundations/open_clip/releases/tag/v3.2.0
---
 .../python-modules/open-clip-torch/default.nix     | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/pkgs/development/python-modules/open-clip-torch/default.nix b/pkgs/development/python-modules/open-clip-torch/default.nix
index 0631206e1e5f..11772f0033d5 100644
--- a/pkgs/development/python-modules/open-clip-torch/default.nix
+++ b/pkgs/development/python-modules/open-clip-torch/default.nix
@@ -24,19 +24,17 @@
   pandas,
   transformers,
   webdataset,
-
-  stdenv,
 }:
 buildPythonPackage rec {
   pname = "open-clip-torch";
-  version = "3.1.0";
+  version = "3.2.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "mlfoundations";
     repo = "open_clip";
     tag = "v${version}";
-    hash = "sha256-xDXxhncX0l9qwbV4Gk4rOROH6Qyit/FSTgjrg4Vbk1s=";
+    hash = "sha256-k4/u0XtfBmPSVKfEK3wHqJXtKAuUNkUnk1TLG2S6PPs=";
   };
 
   build-system = [ pdm-backend ];
@@ -77,14 +75,12 @@ buildPythonPackage rec {
     "test_inference_with_data"
     "test_pretrained_text_encoder"
     "test_training_mt5"
+
     # fails due to type errors
     "test_num_shards"
-  ]
-  ++ lib.optionals (stdenv.hostPlatform.isAarch64 && stdenv.hostPlatform.isLinux) [
+
+    # hangs forever
     "test_training"
-    "test_training_coca"
-    "test_training_unfreezing_vit"
-    "test_training_clip_with_jit"
   ];
 
   meta = {