push sheeet

pkgs/development/python-modules/metaflow/default.nix

@@ -0,0 +1,55 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+  boto3,
+  requests,
+  setuptools,
+  python,
+}:
+
+buildPythonPackage rec {
+  pname = "metaflow";
+  version = "2.18.10";
+  pyproject = true;
+
+  src = fetchFromGitHub {
+    owner = "Netflix";
+    repo = "metaflow";
+    tag = version;
+    hash = "sha256-kDq+w3BWC8ug4PeLhkXe3JUy2pBLaKqceRBqhtkCGN4=";
+  };
+
+  build-system = [
+    setuptools
+  ];
+
+  dependencies = [
+    boto3
+    requests
+  ];
+
+  checkPhase = ''
+    runHook preCheck
+
+    export HOME="$TMPDIR"
+    export USER="metaflow-test-user"
+
+    pushd test/core
+    ${python.interpreter} run_tests.py --num-parallel $NIX_BUILD_CORES \
+      --tests FlowOptionsTest,BasicLogTest
+    popd
+
+    runHook postCheck
+  '';
+
+  pythonImportsCheck = [ "metaflow" ];
+
+  meta = {
+    description = "Open Source AI/ML Platform";
+    homepage = "https://metaflow.org/";
+    changelog = "https://github.com/Netflix/metaflow/releases/tag/${src.tag}";
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [ kr7x ];
+  };
+}