summaryrefslogtreecommitdiffstats
path: root/pkgs/by-name/fa/fasttrackml/package.nix
blob: 14a76456dc82eced1df145cb04af03cc09367766 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
{
  lib,
  buildGoModule,
  fetchFromGitHub,
}:

buildGoModule (finalAttrs: {
  pname = "fasttrackml";
  version = "0.6.0";

  src = fetchFromGitHub {
    owner = "G-Research";
    repo = "fasttrackml";
    tag = "v${finalAttrs.version}";
    hash = "sha256-Z1Hx8nYaifdlnFqp709+rKVXXXMa6cS3+evTG8ZnwrU=";
  };

  vendorHash = "sha256-GPnhE85mHg4KyPeB6+fUP4Y1MlpYTgidqawPddg5kyw=";

  tags = [
    "netgo"
    "osusergo"
    "sqlite_foreign_keys"
    "sqlite_math_functions"
    "sqlite_omit_load_extension"
    "sqlite_unlock_notify"
  ];

  # skip several tests that need network access
  checkFlags =
    let
      skippedTests = [
        # failed to connect to `host=localhost user=postgres database=postgres
        "TestImportTestSuite/Test_Ok/sqlite->postgres"
        "TestImportTestSuite/Test_Ok/sqlcipher->postgres"
        "TestImportTestSuite/Test_Ok/postgres->sqlite"
        "TestImportTestSuite/Test_Ok/postgres->sqlcipher"
        "TestImportTestSuite/Test_Ok/postgres->postgres"

        # test timed out after 10m0s while creating Google Cloud storage bucket
        "TestGetArtifactGSTestSuite/Test_Error"
        "TestGetArtifactGSTestSuite/Test_Ok"
        "TestListArtifactGSTestSuite/Test_Error"
        "TestListArtifactGSTestSuite/Test_Ok"

        # test failure while trying to create/access S3 bucket
        "TestGetArtifactS3TestSuite/Test_Error"
        "TestGetArtifactS3TestSuite/Test_Ok"
        "TestListArtifactS3TestSuite/Test_Error"
        "TestListArtifactS3TestSuite/Test_Ok"

        # connect: network is unreachable
        "TestArtifactFlowTestSuite/Test_Ok/TestCustomNamespaces"
        "TestArtifactFlowTestSuite/Test_Ok/TestExplicitDefaultAndCustomNamespaces"
        "TestArtifactFlowTestSuite/Test_Ok/TestImplicitDefaultAndCustomNamespaces"
      ];
    in
    [ "-skip=^${builtins.concatStringsSep "$|^" skippedTests}$" ];

  postInstall = ''
    # we don't need this binary and it messes with our real python
    rm $out/bin/python
  '';

  meta = {
    description = "API for logging parameters and metrics when running machine learning code";
    homepage = "https://github.com/G-Research/fasttrackml";
    changelog = "https://github.com/G-Research/fasttrackml/releases/tag/v${finalAttrs.version}";
    license = lib.licenses.asl20;
    maintainers = with lib.maintainers; [ jherland ];
    platforms = lib.platforms.unix;
    mainProgram = "fasttrackml";
  };
})