Added boxmot

flake.nix

@@ -23,8 +23,7 @@
             fetchSubmodules = true;
           };
 
-          propagatedBuildInputs =
+          propagatedBuildInputs = [ pythonPackages.numpy ];
-            [ pythonPackages.numpy pythonPackages.cython ];
 
           nativeBuildInputs = [
             pythonPackages.setuptools
@@ -33,12 +32,7 @@
           ];
 
           meta = with pkgs.lib; {
-            description = ''
+            description = "Linear Assignment Problem solver (LAPJV/LAPMOD).";
-              THOP offers an intuitive API to profile PyTorch models
-              by calculating the number of MACs and parameters. This
-              functionality is crucial for assessing the computational
-              efficiency and memory footprint of deep learning models.
-            '';
             homepage = "https://github.com/rathaROG/lapx";
             license = pkgs.lib.licenses.mit;
             maintainers = with maintainers; [ ];
@@ -127,7 +121,56 @@
           };
         };
 
-        env = pkgs.python312.withPackages (ps: [ lapx ]);
+        # boxmot
+        boxmot = pkgs.python3Packages.buildPythonPackage rec {
+          version = "0.5.11";
+          pname = "lapx";
+          pyproject = true;
+          src = pkgs.fetchFromGitHub {
+            owner = "mikel-brostrom";
+            repo = "boxmot";
+            rev = "5cbcd86b2716db5c70519b41e12907d351d7f5ff";
+            hash = "sha256-fb5qRdR75szHpFwil/uJz3/KAutYtrDVjyYJRZhI3pA=";
+            fetchSubmodules = true;
+          };
+
+          propagatedBuildInputs = [
+            pythonPackages.loguru
+            pythonPackages.numpy
+            pythonPackages.pandas
+            pythonPackages.pyyaml
+            pythonPackages.torch
+            pythonPackages.yacs
+            pythonPackages.torchvision
+            pythonPackages.ftfy
+            pythonPackages.filterpy
+            pythonPackages.gdown
+            pythonPackages.regex
+            pythonPackages.scikit-learn
+            pythonPackages.gitpython
+            pythonPackages.opencv4
+            pythonPackages.pip
+            lapx
+          ];
+
+          pythonRelaxDeps = [ "torchvision" ];
+          pythonRemoveDeps = [ "opencv-python" ];
+
+          nativeBuildInputs = [
+            pythonPackages.setuptools
+            pythonPackages.setuptools-scm
+            pythonPackages.poetry-core
+          ];
+
+          meta = with pkgs.lib; {
+            description = "Linear Assignment Problem solver (LAPJV/LAPMOD).";
+            homepage = "https://github.com/mikel-brostrom/boxmot";
+            license = pkgs.lib.licenses.mit;
+            maintainers = with maintainers; [ ];
+          };
+        };
+
+        env = pkgs.python312.withPackages (ps: [ boxmot ]);
 
       in {
         packages = {