Loading
+21 −8
Original line number Diff line number Diff line
@@ -6,16 +6,22 @@

python3.pkgs.buildPythonApplication rec {
  pname = "dbx";
  version = "0.7.6";
  version = "0.8.8";
  format = "setuptools";

  src = fetchFromGitHub {
    owner = "databrickslabs";
    repo = "dbx";
    rev = "refs/tags/v${version}";
    hash = "sha256-P/cniy0xYaDoUbKdvV7KCubCpmOAhYp3cg2VBRA+a6I=";
    hash = "sha256-nx6fz+atlnB/KxdznnZArHpyv41cuBDQauG0irq1Zyc=";
  };

  postPatch = ''
    substituteInPlace setup.py \
      --replace "mlflow-skinny>=1.28.0,<3.0.0" "mlflow" \
      --replace "rich==12.6.0" "rich"
  '';

  propagatedBuildInputs = with python3.pkgs; [
    aiohttp
    click
@@ -34,6 +40,19 @@ python3.pkgs.buildPythonApplication rec {
    watchdog
  ] ++ typer.optional-dependencies.all;

  passthru.optional-dependencies = with python3.pkgs; {
    aws = [
      boto3
    ];
    azure = [
      azure-storage-blob
      azure-identity
    ];
    gcp = [
      google-cloud-storage
    ];
  };

  nativeCheckInputs = [
    git
  ] ++ (with python3.pkgs; [
@@ -43,12 +62,6 @@ python3.pkgs.buildPythonApplication rec {
    pytestCheckHook
  ]);

  postPatch = ''
    substituteInPlace setup.py \
      --replace "mlflow-skinny>=1.28.0,<=2.0.0" "mlflow" \
      --replace "rich==12.5.1" "rich"
  '';

  preCheck = ''
    export HOME=$(mktemp -d)
    export PATH="$PATH:$out/bin"