Commit bbc563dd authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent 876b90e2
Loading
Loading
Loading
Loading
+9 −6
Original line number Diff line number Diff line
{
  lib,
  buildPythonPackage,
  fetchFromGitHub,

  # build-system
@@ -7,12 +8,13 @@

  # dependencies
  alembic,
  buildPythonPackage,
  cachetools,
  click,
  cloudpickle,
  cryptography,
  databricks-sdk,
  docker,
  fastapi,
  flask,
  gitpython,
  graphene,
@@ -34,6 +36,7 @@
  scipy,
  sqlalchemy,
  sqlparse,
  uvicorn,

  # tests
  aiohttp,
@@ -44,7 +47,6 @@
  botocore,
  catboost,
  datasets,
  fastapi,
  google-cloud-storage,
  httpx,
  jwt,
@@ -65,20 +67,19 @@
  tensorflow,
  torch,
  transformers,
  uvicorn,
  xgboost,
}:

buildPythonPackage rec {
  pname = "mlflow";
  version = "3.1.4";
  version = "3.3.1";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "mlflow";
    repo = "mlflow";
    tag = "v${version}";
    hash = "sha256-GU1CVwc31bLai8QoXch/UTx//+OG2V/6jG9Ocz4MDuA=";
    hash = "sha256-5zObSnGx7+cCrqRfvcnprQN05NqVBCeWcAZEE1Jpeuo=";
  };

  pythonRelaxDeps = [
@@ -97,8 +98,10 @@ buildPythonPackage rec {
    cachetools
    click
    cloudpickle
    cryptography
    databricks-sdk
    docker
    fastapi
    flask
    gitpython
    graphene
@@ -122,6 +125,7 @@ buildPythonPackage rec {
    shap
    sqlalchemy
    sqlparse
    uvicorn
  ];

  pythonImportsCheck = [ "mlflow" ];
@@ -135,7 +139,6 @@ buildPythonPackage rec {
    botocore
    catboost
    datasets
    fastapi
    google-cloud-storage
    httpx
    jwt