From 6add9f882954b20287d54009b9c61c4ff6cadd79 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Sun, 21 May 2023 00:46:04 +0200
Subject: [PATCH] python310Packages.timm: 0.6.12 -> 0.9.2

Diff: https://github.com/huggingface/pytorch-image-models/compare/refs/tags/v0.6.12...v0.9.2

Changelog: https://github.com/huggingface/pytorch-image-models/blob/v0.9.2/README.md#whats-new
---
 .../python-modules/timm/default.nix           | 33 +++++++++++++++----
 1 file changed, 26 insertions(+), 7 deletions(-)

diff --git a/pkgs/development/python-modules/timm/default.nix b/pkgs/development/python-modules/timm/default.nix
index e93563d69b4..e67ed91c135 100644
--- a/pkgs/development/python-modules/timm/default.nix
+++ b/pkgs/development/python-modules/timm/default.nix
@@ -7,33 +7,52 @@
 , pytest-timeout
 , huggingface-hub
 , pyyaml
+, safetensors
 , torch
 , torchvision
 }:
 
 buildPythonPackage rec {
   pname = "timm";
-  version = "0.6.12";
-  disabled = pythonOlder "3.6";
+  version = "0.9.2";
+  format = "setuptools";
+
+  disabled = pythonOlder "3.7";
 
   src = fetchFromGitHub {
     owner = "huggingface";
     repo = "pytorch-image-models";
     rev = "refs/tags/v${version}";
-    hash = "sha256-RNjCcCnNhtr5a+29Bx+k427a03MSooqvnuiDQ8cT8FA=";
+    hash = "sha256-gYrc8ds6urZvwDsTnzPjxjSTiAGzUD3RlCf0wogCrDI=";
   };
 
   propagatedBuildInputs = [
     huggingface-hub
     pyyaml
+    safetensors
     torch
     torchvision
   ];
 
-  nativeCheckInputs = [ expecttest pytestCheckHook pytest-timeout ];
-  pytestFlagsArray = [ "tests" ];
-  # takes too long and also tries to download models:
-  disabledTestPaths = [ "tests/test_models.py" ];
+  nativeCheckInputs = [
+    expecttest
+    pytestCheckHook
+    pytest-timeout
+  ];
+
+  pytestFlagsArray = [
+    "tests"
+  ];
+
+  disabledTestPaths = [
+    # Takes too long and also tries to download models
+    "tests/test_models.py"
+  ];
+
+  disabledTests = [
+    # AttributeError: 'Lookahead' object has no attribute '_optimizer_step_pre...
+    "test_lookahead"
+  ];
 
   pythonImportsCheck = [
     "timm"