1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 pythonOlder,
6 setuptools,
7 wheel,
8}:
9
10buildPythonPackage rec {
11 pname = "dllogger";
12 version = "1.1.0";
13 pyproject = true;
14
15 disabled = pythonOlder "3.5";
16
17 src = fetchFromGitHub {
18 owner = "NVIDIA";
19 repo = "dllogger";
20 tag = "v${version}";
21 hash = "sha256-kT6FhAl4JZlFPdzKYopAJBurYVMaU5umn/qZADfPjkE=";
22 };
23
24 nativeBuildInputs = [
25 setuptools
26 wheel
27 ];
28
29 # use examples as smoke tests since upstream has no tests
30 checkPhase = ''
31 runHook preCheck
32
33 python examples/dllogger_example.py
34 python examples/dllogger_singleton_example.py
35
36 runHook postCheck
37 '';
38
39 pythonImportsCheck = [ "dllogger" ];
40
41 meta = with lib; {
42 description = "Logging tool for deep learning";
43 homepage = "https://github.com/NVIDIA/dllogger";
44 changelog = "https://github.com/NVIDIA/dllogger/releases/tag/${src.tag}";
45 license = licenses.asl20;
46 maintainers = with maintainers; [ natsukium ];
47 };
48}