From 703675d2189aa2d1b101f4e5297a16e6e9845cd9 Mon Sep 17 00:00:00 2001
From: David Crompton <david.crompton@mail.utoronto.ca>
Date: Thu, 18 Jul 2024 19:07:10 -0400
Subject: [PATCH] KCNHUB: Python: Overlays for MNE & EDFIO

---
 overlays/python-packages.nix | 26 +++++++++++++++++++++++++-
 overlays/python.nix          |  2 +-
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/overlays/python-packages.nix b/overlays/python-packages.nix
index 3218c59..1d7a21f 100644
--- a/overlays/python-packages.nix
+++ b/overlays/python-packages.nix
@@ -1,4 +1,4 @@
-final-py: prev-py: {
+final: final-py: prev-py: {
   mne-python = prev-py.mne-python.overridePythonAttrs (old: {
     pyproject = true;
     format = null;
@@ -6,4 +6,28 @@ final-py: prev-py: {
       final-py.setuptools-scm
     ];
   });
+
+  edfio = let
+    version = "0.4.1";
+    pname = "edfio";
+  in final-py.buildPythonPackage {
+    inherit pname version;
+    format = "pyproject";
+    
+    src = final.fetchFromGitHub {
+      owner = "the-siesta-group";
+      repo = pname;
+      rev = "v${version}";
+      hash = "sha256-+yb3Q/JximkVS6SOTFnjMYghbOi+UBbqW2Bs29eSonA=";
+    };
+
+    nativeBuildInputs = with final-py; [
+      poetry-core
+      poetry-dynamic-versioning
+    ];
+    
+    propagatedBuildInputs = with final-py; [
+      numpy
+    ];
+  };
 }
diff --git a/overlays/python.nix b/overlays/python.nix
index f5208fe..1c8d118 100644
--- a/overlays/python.nix
+++ b/overlays/python.nix
@@ -1,5 +1,5 @@
 final: prev: {
   pythonPackagesExtensions = prev.pythonPackagesExtensions ++ [(
-    (import ./python-packages.nix)
+    (import ./python-packages.nix final)
   )];
 }