From 74561ffeac40631621af34a6d469c6571a8cdf23 Mon Sep 17 00:00:00 2001
From: Dennis <dennish@wuitz.de>
Date: Mon, 21 Oct 2024 17:09:03 +0200
Subject: [PATCH] python3Packages.torch-tb-profiler: 0.3.1 -> 0.4.0

---
 .../torch-tb-profiler/default.nix             | 30 +++++++++++--------
 1 file changed, 18 insertions(+), 12 deletions(-)

diff --git a/pkgs/development/python-modules/torch-tb-profiler/default.nix b/pkgs/development/python-modules/torch-tb-profiler/default.nix
index 005c9f375455..766baff5d3cc 100644
--- a/pkgs/development/python-modules/torch-tb-profiler/default.nix
+++ b/pkgs/development/python-modules/torch-tb-profiler/default.nix
@@ -1,32 +1,34 @@
 {
+  lib,
   buildPythonPackage,
   fetchFromGitHub,
-  lib,
   pandas,
   pytestCheckHook,
-  torch,
+  setuptools,
   tensorboard,
+  torch,
   torchvision,
 }:
-
 let
-  version = "0.3.1";
+  version = "0.4.0";
   repo = fetchFromGitHub {
     owner = "pytorch";
     repo = "kineto";
-    rev = "v${version}";
-    hash = "sha256-Yg001XzOPDmz9wEP2b7Ggz/uU6x5PFzaaBeUBwWKFS0=";
+    rev = "refs/tags/v${version}";
+    hash = "sha256-nAtqGCv8q3Tati3NOGWWLb+gXdvO3qmECeC1WG2Mt3M=";
   };
 in
-buildPythonPackage rec {
+buildPythonPackage {
   pname = "torch_tb_profiler";
   inherit version;
-  format = "setuptools";
+  pyproject = true;
 
   # See https://discourse.nixos.org/t/extracting-sub-directory-from-fetchgit-or-fetchurl-or-any-derivation/8830.
   src = "${repo}/tb_plugin";
 
-  propagatedBuildInputs = [
+  build-system = [ setuptools ];
+
+  dependencies = [
     pandas
     tensorboard
   ];
@@ -43,14 +45,18 @@ buildPythonPackage rec {
     "test_tensorboard_end2end"
     "test_tensorboard_with_path_prefix"
     "test_tensorboard_with_symlinks"
+    "test_autograd_api"
+    "test_profiler_api_with_record_shapes_memory_stack"
+    "test_profiler_api_without_record_shapes_memory_stack"
+    "test_profiler_api_without_step"
   ];
 
   pythonImportsCheck = [ "torch_tb_profiler" ];
 
-  meta = with lib; {
+  meta = {
     description = "PyTorch Profiler TensorBoard Plugin";
     homepage = "https://github.com/pytorch/kineto";
-    license = licenses.bsd3;
-    maintainers = with maintainers; [ samuela ];
+    license = lib.licenses.bsd3;
+    maintainers = with lib.maintainers; [ samuela ];
   };
 }