pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 pytestCheckHook,
6 setuptools,
7 numpy,
8 torch,
9}:
10
11buildPythonPackage rec {
12 pname = "roma";
13 version = "1.5.4";
14 pyproject = true;
15
16 src = fetchFromGitHub {
17 owner = "naver";
18 repo = "roma";
19 tag = "v${version}";
20 hash = "sha256-byPW58I+6mCE2fR6eVNQfNDCLbZSfoPmPbc/GuRpKGo=";
21 };
22
23 build-system = [
24 setuptools
25 ];
26
27 dependencies = [
28 numpy
29 torch
30 ];
31
32 nativeCheckInputs = [
33 pytestCheckHook
34 ];
35
36 pythonImportsCheck = [
37 "roma"
38 ];
39
40 meta = {
41 description = "Lightweight library to deal with 3D rotations in PyTorch";
42 homepage = "https://github.com/naver/roma";
43 license = lib.licenses.bsd3;
44 maintainers = with lib.maintainers; [ nim65s ];
45 };
46}