KCNHUB: Python: Add Tonic

machines/kcnhub/packages/python.nix

@@ -43,6 +43,8 @@
       brian2
       # SpiNNaker simulator
       spynnaker
+      # fetching neuromorphic data:
+      tonic
 
       # Machine learning toolkits:
       tensorflow

overlays/python-packages.nix

@@ -30,4 +30,100 @@ final: final-py: prev-py: {
       numpy
     ];
   };
+
+  importRosbag = final-py.buildPythonPackage {
+    pname = "importRosbag";
+    version = "1.0.4.1";
+    
+    src = final.fetchFromGitHub {
+      owner = "event-driven-robotics";
+      repo = "importRosbag";
+      rev = "71183fe375ec0f79512100db16c1a955b7aadf62";
+      hash = "sha256-2ey+IGKqkvvHhEneoZlSx1g074/WKpWKZED04BVXkaM=";
+    };
+
+    propagatedBuildInputs = with final-py; [
+      tqdm
+      numpy
+    ];
+  };
+  
+  expelliarmus = final-py.buildPythonPackage {
+    pname = "expelliarmus";
+    version = "14-03-23";
+    
+    src = final.fetchFromGitHub {
+      owner = "open-neuromorphic";
+      repo = "expelliarmus";
+      rev = "6ed9e41e4d289b45d5e2ca2526cff4ea276f3494";
+      hash = "sha256-glW0W6OjTpWu+DKtGHBaJXBm7XMZSQYEaC2DjghQFho=";
+    };
+
+    buildInputs = with final-py; [
+      pytest
+    ];
+
+    propagatedBuildInputs = with final-py; [
+      numpy
+    ];
+  };
+  
+  pytorch-data = final-py.buildPythonPackage rec {
+    pname = "pytorch-data";
+    version = "0.7.1";
+
+    src = final.fetchFromGitHub {
+      owner = "pytorch";
+      repo = "data";
+      rev = "v${version}";
+      hash = "sha256-yF3WHVOP1BIhGpitkQDy8dry4ZnEOh4+U6XTaG7mBv4=";
+      fetchSubmodules = true;
+    };
+
+    doCheck = false;
+
+    nativeBuildInputs = with final; [
+      cmake
+    ];
+
+    dontUseCmakeConfigure = true;
+    
+    buildInputs = with final; [
+      git
+    ];
+
+    propagatedBuildInputs = with final-py; [
+      urllib3
+      requests
+      torch
+    ];
+  };
+
+  tonic = final-py.buildPythonPackage rec {
+    pname = "tonic";
+    version = "1.5.0";
+    env.PBR_VERSION = version;
+    src = final.fetchFromGitHub {
+      owner = "neuromorphs";
+      repo = "tonic";
+      rev = "v${version}";
+      hash = "sha256-edwd/iJpwBXGw7CB7bD/ZVW/47pwrTOzD2lX5ULHiMY=";
+    };
+
+    propagatedBuildInputs = with final-py; [
+      numpy
+      h5py
+      importRosbag
+      scipy
+      tqdm
+      librosa
+      pbr
+      expelliarmus
+      typing-extensions
+      # Technically not required
+      torch
+      torchaudio
+      pytorch-data
+    ];
+  };
 } // (import ./spynnaker.nix final final-py prev-py)