Unverified Commit 07e7a919 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.wandb: 0.21.4 -> 0.24.0 (#482115)

parents d859f82c b95c92c9
Loading
Loading
Loading
Loading
+18 −7
Original line number Diff line number Diff line
@@ -34,18 +34,19 @@
  # tests
  pytestCheckHook,
  writableTmpDirAsHomeHook,
  pytest-timeout,
}:

buildPythonPackage rec {
buildPythonPackage (finalAttrs: {
  pname = "lerobot";
  version = "0.4.0";
  version = "0.4.2";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "huggingface";
    repo = "lerobot";
    tag = "v${version}";
    hash = "sha256-RVe1X0qBPm+okO3Gi/UdkuvuX0m4RlbhIs+NJLlC9wU=";
    tag = "v${finalAttrs.version}";
    hash = "sha256-qBbXFvVQ+cESBrNy8NIoT6GR6dqzeLbRNe3JmcpiiTw=";
  };

  # ValueError: mutable default <class 'lerobot.configs.types.PolicyFeature'> for field value is not allowed: use default_factory
@@ -67,12 +68,12 @@ buildPythonPackage rec {

  pythonRelaxDeps = [
    "av"
    "datasets"
    "draccus"
    "gymnasium"
    "rerun-sdk"
    "torch"
    "torchvision"
    "wandb"
  ];

  dependencies = [
@@ -109,6 +110,7 @@ buildPythonPackage rec {
  nativeCheckInputs = [
    pytestCheckHook
    writableTmpDirAsHomeHook
    pytest-timeout
  ];

  disabledTests = [
@@ -124,6 +126,9 @@ buildPythonPackage rec {
    "test_dataset_initialization"
    "test_factory"
    "test_from_pretrained_nonexistent_path"
    "test_load_config_nonexistent_path_tries_hub"
    "test_make_env_from_hub_async"
    "test_make_env_from_hub_with_trust"
    "test_policy_defaults"
    "test_save_and_load_pretrained"

@@ -149,18 +154,24 @@ buildPythonPackage rec {
    #  Regex: "Can't instantiate abstract class NonCallableStep with abstract method __call_"
    #  Input: "Can't instantiate abstract class NonCallableStep without an implementation for abstract method '__call__'"
    "test_construction_rejects_step_without_call"

    # TypeError: 'NoneType' object is not subscriptable
    "test_pi0_rtc_inference_with_prev_chunk"
  ];

  disabledTestPaths = [
    # Sometimes hang forever on some CPU models
    "tests/policies/test_sac_policy.py"

    # Sometimes hang forever
    "tests/policies/rtc/test_modeling_rtc.py"
  ];

  meta = {
    description = "Making AI for Robotics more accessible with end-to-end learning";
    homepage = "https://github.com/huggingface/lerobot";
    changelog = "https://github.com/huggingface/lerobot/releases/tag/v${version}";
    changelog = "https://github.com/huggingface/lerobot/releases/tag/${finalAttrs.src.tag}";
    license = lib.licenses.asl20;
    maintainers = with lib.maintainers; [ GaetanLepage ];
  };
}
})
+15 −14
Original line number Diff line number Diff line
@@ -2,10 +2,12 @@
  lib,
  stdenv,
  fetchFromGitHub,
  pythonAtLeast,

  ## wandb-core
  buildGo125Module,
  buildGoModule,
  gitMinimal,
  writableTmpDirAsHomeHook,
  versionCheckHook,

  ## gpu-stats
@@ -20,11 +22,9 @@

  # dependencies
  click,
  docker-pycreds,
  gitpython,
  platformdirs,
  protobuf,
  psutil,
  pydantic,
  pyyaml,
  requests,
@@ -72,16 +72,15 @@
  torch,
  torchvision,
  tqdm,
  writableTmpDirAsHomeHook,
}:

let
  version = "0.21.4";
  version = "0.24.0";
  src = fetchFromGitHub {
    owner = "wandb";
    repo = "wandb";
    tag = "v${version}";
    hash = "sha256-1l68nU/rmYg/Npg1EVraGr2tu/lkNAo9M7Q0IyckEoc=";
    hash = "sha256-dICa/sIFEHI59gJxrvWyI9Uc3rbwXi+Xh60O/hElZh0=";
  };

  gpu-stats = rustPlatform.buildRustPackage {
@@ -108,7 +107,7 @@ let
    };
  };

  wandb-core = buildGo125Module rec {
  wandb-core = buildGoModule rec {
    pname = "wandb-core";
    inherit src version;

@@ -126,6 +125,7 @@ let

    nativeBuildInputs = [
      gitMinimal
      writableTmpDirAsHomeHook
    ];

    nativeInstallCheckInputs = [
@@ -151,7 +151,7 @@ let
  };
in

buildPythonPackage rec {
buildPythonPackage (finalAttrs: {
  pname = "wandb";
  pyproject = true;

@@ -193,11 +193,9 @@ buildPythonPackage rec {

  dependencies = [
    click
    docker-pycreds
    gitpython
    platformdirs
    protobuf
    psutil
    pydantic
    pyyaml
    requests
@@ -382,16 +380,19 @@ buildPythonPackage rec {

    # Breaks in sandbox: "Timed out waiting for wandb service to start"
    "test_setup_offline"
  ]
  ++ lib.optionals (pythonAtLeast "3.14") [
    # AttributeError: '...' object has no attribute '__annotations__'
    "test_watch_graph_torch_jit"
    "test_watch_parameters_torch_jit"
  ];

  pythonImportsCheck = [ "wandb" ];

  meta = {
    description = "CLI and library for interacting with the Weights and Biases API";
    homepage = "https://github.com/wandb/wandb";
    changelog = "https://github.com/wandb/wandb/raw/v${version}/CHANGELOG.md";
    changelog = "https://github.com/wandb/wandb/raw/${finalAttrs.version}/CHANGELOG.md";
    license = lib.licenses.mit;
    maintainers = with lib.maintainers; [ samuela ];
    broken = gpu-stats.meta.broken || wandb-core.meta.broken;
  };
}
})