nur-packages

commit fecf5fb2bdc6c1d18904ab821a6dcaa3026f3416
parent abfd221c12daf678ebf829fe96acd038de188aea
Author: Nikolay Korotkiy <sikmir@disroot.org>
Date:   Sat, 15 Aug 2026 18:02:27 +0400

Add torchgeo

Diffstat:
A
pkgs/by-name/to/torchgeo/package.nix
 | 
68
++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 68 insertions(+), 0 deletions(-)
diff --git a/pkgs/by-name/to/torchgeo/package.nix b/pkgs/by-name/to/torchgeo/package.nix
@@ -0,0 +1,68 @@
+{
+  lib,
+  fetchFromGitHub,
+  python3Packages,
+  lightly,
+  segmentation-models-pytorch,
+  writableTmpDirAsHomeHook,
+}:
+
+python3Packages.buildPythonPackage (finalAttrs: {
+  pname = "torchgeo";
+  version = "0.10.0";
+  pyproject = true;
+
+  __structuredAttrs = true;
+
+  src = fetchFromGitHub {
+    owner = "torchgeo";
+    repo = "torchgeo";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-Uvx92HCQyCSjgdQPOtpHwvx4irr7kJsfs481Xl/mEEI=";
+  };
+
+  build-system = with python3Packages; [ uv-build ];
+
+  dependencies =
+    with python3Packages;
+    [
+      einops
+      geopandas
+      jsonargparse
+      kornia
+      lightly
+      lightning
+      matplotlib
+      numpy
+      pandas
+      pillow
+      pyogrio
+      pyproj
+      rasterio
+      requests
+      segmentation-models-pytorch
+      shapely
+      timm
+      torch
+      torchmetrics
+      tqdm
+      typing-extensions
+    ]
+    ++ jsonargparse.optional-dependencies.jsonnet
+    ++ jsonargparse.optional-dependencies.signatures;
+
+  nativeCheckInputs = with python3Packages; [
+    pytestCheckHook
+    pytest-socket
+    writableTmpDirAsHomeHook
+  ];
+
+  doCheck = false;
+
+  meta = {
+    description = "TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data";
+    homepage = "https://github.com/torchgeo/torchgeo";
+    license = lib.licenses.mit;
+    maintainers = [ lib.maintainers.sikmir ];
+  };
+})