package.nix (1385B)
1 { 2 lib, 3 fetchFromGitHub, 4 python3Packages, 5 lightly, 6 segmentation-models-pytorch, 7 writableTmpDirAsHomeHook, 8 }: 9 10 python3Packages.buildPythonPackage (finalAttrs: { 11 pname = "torchgeo"; 12 version = "0.10.0"; 13 pyproject = true; 14 15 __structuredAttrs = true; 16 17 src = fetchFromGitHub { 18 owner = "torchgeo"; 19 repo = "torchgeo"; 20 tag = "v${finalAttrs.version}"; 21 hash = "sha256-Uvx92HCQyCSjgdQPOtpHwvx4irr7kJsfs481Xl/mEEI="; 22 }; 23 24 build-system = with python3Packages; [ uv-build ]; 25 26 dependencies = 27 with python3Packages; 28 [ 29 einops 30 geopandas 31 jsonargparse 32 kornia 33 lightly 34 lightning 35 matplotlib 36 numpy 37 pandas 38 pillow 39 pyogrio 40 pyproj 41 rasterio 42 requests 43 segmentation-models-pytorch 44 shapely 45 timm 46 torch 47 torchmetrics 48 tqdm 49 typing-extensions 50 ] 51 ++ jsonargparse.optional-dependencies.jsonnet 52 ++ jsonargparse.optional-dependencies.signatures; 53 54 nativeCheckInputs = with python3Packages; [ 55 pytestCheckHook 56 pytest-socket 57 writableTmpDirAsHomeHook 58 ]; 59 60 doCheck = false; 61 62 meta = { 63 description = "TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data"; 64 homepage = "https://github.com/torchgeo/torchgeo"; 65 license = lib.licenses.mit; 66 maintainers = [ lib.maintainers.sikmir ]; 67 }; 68 })