From e7615431d9ab3c80385da9e8ace6502742400e0d Mon Sep 17 00:00:00 2001
From: movefasta <movefasta@dezcom.org>
Date: Sun, 28 May 2023 22:55:43 +0300
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=20webdataset,=20=D0=BD=D0=B5=D0=BE=D0=B1=D1=85=D0=BE?=
 =?UTF-8?q?=D0=B4=D0=B8=D0=BC=D1=8B=D0=B9=20=D0=B4=D0=BB=D1=8F=20megapose6?=
 =?UTF-8?q?d?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 flake.nix                                     |  6 +--
 .../python-modules/megapose6d/default.nix     |  3 +-
 .../python-modules/webdataset/default.nix     | 44 +++++++++++++++++++
 3 files changed, 49 insertions(+), 4 deletions(-)
 create mode 100644 pkgs/development/python-modules/webdataset/default.nix

diff --git a/flake.nix b/flake.nix
index 2647b5a..e929e5e 100644
--- a/flake.nix
+++ b/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;
           };
 
diff --git a/pkgs/development/python-modules/megapose6d/default.nix b/pkgs/development/python-modules/megapose6d/default.nix
index a4d575b..1b19f6e 100644
--- a/pkgs/development/python-modules/megapose6d/default.nix
+++ b/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
   ];
diff --git a/pkgs/development/python-modules/webdataset/default.nix b/pkgs/development/python-modules/webdataset/default.nix
new file mode 100644
index 0000000..7637ea2
--- /dev/null
+++ b/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; [ ];
+  };
+}