Добавлен webdataset, необходимый для megapose6d

flake.nix

@@ -36,9 +36,9 @@
           fstl = pkgs.libsForQt5.callPackage ./pkgs/misc/fstl { };
           open3d = pkgs.callPackage ./open3d.nix { };
           unified-planning = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/unified-planning { };
-          tamer = pkgs.python3Packages.callPackage ./tamer.nix { };
           blenderproc = pkgs.python3Packages.callPackage ./blenderproc.nix { };
-          megapose6d = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/megapose6d { };
+          webdataset = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/webdataset { };
+          megapose6d = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/megapose6d { webdataset = webdataset; };
           slic3r = pkgs.callPackage ./pkgs/misc/slic3r { };
         in
         {
@@ -52,7 +52,7 @@
           };
 
           packages = {
-            inherit mayo blender bpy pysdf fstl darknet unified-planning slic3r;
+            inherit mayo blender bpy pysdf fstl darknet unified-planning slic3r megapose6d;
             gazebo = ros.humble.gazebo;
           };
 

pkgs/development/python-modules/megapose6d/default.nix

@@ -2,6 +2,7 @@
 , buildPythonPackage
 , python3Packages
 , fetchFromGitHub
+, webdataset
 }:
 
 buildPythonPackage rec {
@@ -38,7 +39,7 @@ buildPythonPackage rec {
     simplejson
     pyroma
     seaborn
-    # webdataset
+    webdataset
     # panda3d
     # open3d
   ];

pkgs/development/python-modules/webdataset/default.nix

@@ -0,0 +1,44 @@
+{ lib
+, buildPythonPackage
+, fetchFromGitHub
+, python3
+, braceexpand
+, dataclasses-json
+, numpy
+, pyyaml
+}:
+
+buildPythonPackage rec {
+  pname = "webdataset";
+  version = "0.2.49";
+  format = "setuptools";
+
+  src = fetchFromGitHub {
+    owner = "webdataset";
+    repo = "webdataset";
+    rev = version;
+    hash = "sha256-ZIe3fjtilMsY4fCG7E1OWERFFjFg86KYTTEq/dZDjOM=";
+  };
+
+  propagatedBuildInputs = [
+    braceexpand
+    dataclasses-json
+    numpy
+    pyyaml
+  ];
+
+  pythonImportsCheck = [ "webdataset" ];
+
+  checkPhase = ''
+    runHook preCheck
+    ${python3.interpreter} -m unittest
+    runHook postCheck
+  '';
+
+  meta = with lib; {
+    description = "A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch";
+    homepage = "https://github.com/webdataset/webdataset";
+    license = licenses.bsd3;
+    maintainers = with maintainers; [ ];
+  };
+}