diff --git a/machines/kcnhub/packages/python.nix b/machines/kcnhub/packages/python.nix index f271096..d623b4f 100644 --- a/machines/kcnhub/packages/python.nix +++ b/machines/kcnhub/packages/python.nix @@ -43,6 +43,8 @@ brian2 # SpiNNaker simulator spynnaker + # fetching neuromorphic data: + tonic # Machine learning toolkits: tensorflow diff --git a/overlays/python-packages.nix b/overlays/python-packages.nix index 9f10fec..811d3d5 100644 --- a/overlays/python-packages.nix +++ b/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)