diff options
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/wandb/default.nix')
| -rw-r--r-- | nixpkgs/pkgs/development/python-modules/wandb/default.nix | 30 |
1 files changed, 19 insertions, 11 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/wandb/default.nix b/nixpkgs/pkgs/development/python-modules/wandb/default.nix index 25f7e4b98dd9..0c06b4989479 100644 --- a/nixpkgs/pkgs/development/python-modules/wandb/default.nix +++ b/nixpkgs/pkgs/development/python-modules/wandb/default.nix @@ -29,7 +29,6 @@ , pandas , parameterized , pathtools -, promise , protobuf , psutil , pydantic @@ -46,17 +45,15 @@ , sentry-sdk , setproctitle , setuptools -, shortuuid , substituteAll -, tensorflow , torch , tqdm }: buildPythonPackage rec { pname = "wandb"; - version = "0.15.5"; - format = "setuptools"; + version = "0.15.10"; + format = "pyproject"; disabled = pythonOlder "3.6"; @@ -64,7 +61,7 @@ buildPythonPackage rec { owner = pname; repo = pname; rev = "refs/tags/v${version}"; - hash = "sha256-etS1NkkskA5Lg/38QIdzCVWgqZpjpT2LwaWF1k7WVXs="; + hash = "sha256-MuYaeg7+lMOOSalnLyKsCw+f44daDDayvyKvY8z697c="; }; patches = [ @@ -77,6 +74,7 @@ buildPythonPackage rec { nativeBuildInputs = [ pythonRelaxDepsHook + setuptools ]; # setuptools is necessary since pkg_resources is required at runtime. @@ -86,7 +84,6 @@ buildPythonPackage rec { docker_pycreds gitpython pathtools - promise protobuf psutil pyyaml @@ -94,7 +91,6 @@ buildPythonPackage rec { sentry-sdk setproctitle setuptools - shortuuid ]; __darwinAllowLocalNetworking = true; @@ -128,7 +124,6 @@ buildPythonPackage rec { pytestCheckHook responses scikit-learn - tensorflow torch tqdm ]; @@ -141,6 +136,11 @@ buildPythonPackage rec { pythonRelaxDeps = [ "protobuf" ]; + pytestFlagsArray = [ + # We want to run only unit tests + "tests/pytest_tests" + ]; + disabledTestPaths = [ # Tests that try to get chatty over sockets or spin up servers, not possible in the nix build environment. "tests/pytest_tests/system_tests/test_notebooks/test_notebooks.py" @@ -167,7 +167,6 @@ buildPythonPackage rec { "tests/pytest_tests/unit_tests_old/tests_launch/test_launch_aws.py" "tests/pytest_tests/unit_tests_old/tests_launch/test_launch_cli.py" "tests/pytest_tests/unit_tests_old/tests_launch/test_launch_docker.py" - "tests/pytest_tests/unit_tests_old/tests_launch/test_launch_kubernetes.py" "tests/pytest_tests/unit_tests_old/tests_launch/test_launch.py" "tests/pytest_tests/unit_tests/test_cli.py" "tests/pytest_tests/unit_tests/test_data_types.py" @@ -209,6 +208,7 @@ buildPythonPackage rec { "tests/pytest_tests/system_tests/test_core/test_time_resolution.py" "tests/pytest_tests/system_tests/test_core/test_torch_full.py" "tests/pytest_tests/system_tests/test_core/test_validation_data_logger.py" + "tests/pytest_tests/system_tests/test_core/test_wandb_init.py" "tests/pytest_tests/system_tests/test_core/test_wandb_integration.py" "tests/pytest_tests/system_tests/test_core/test_wandb_run.py" "tests/pytest_tests/system_tests/test_core/test_wandb_settings.py" @@ -216,6 +216,7 @@ buildPythonPackage rec { "tests/pytest_tests/system_tests/test_core/test_wandb_verify.py" "tests/pytest_tests/system_tests/test_core/test_wandb.py" "tests/pytest_tests/system_tests/test_importers/test_import_mlflow.py" + "tests/pytest_tests/system_tests/test_nexus/test_nexus.py" "tests/pytest_tests/system_tests/test_sweep/test_public_api.py" "tests/pytest_tests/system_tests/test_sweep/test_sweep_scheduler.py" "tests/pytest_tests/system_tests/test_sweep/test_sweep_utils.py" @@ -230,6 +231,7 @@ buildPythonPackage rec { "tests/pytest_tests/system_tests/test_launch/test_launch_kubernetes.py" "tests/pytest_tests/system_tests/test_launch/test_launch_local_container.py" "tests/pytest_tests/system_tests/test_launch/test_launch_run.py" + "tests/pytest_tests/system_tests/test_launch/test_launch_sagemaker.py" "tests/pytest_tests/system_tests/test_launch/test_launch_sweep_cli.py" "tests/pytest_tests/system_tests/test_launch/test_launch_sweep.py" "tests/pytest_tests/system_tests/test_launch/test_launch.py" @@ -243,7 +245,7 @@ buildPythonPackage rec { "tests/pytest_tests/unit_tests_old/tests_launch/test_launch_jobs.py" # Requires google-cloud-aiplatform which is not packaged as of 2023-04-25. - "tests/pytest_tests/unit_tests_old/tests_launch/test_launch_gcp.py" + "tests/pytest_tests/unit_tests/test_launch/test_runner/test_vertex.py" # Requires google-cloud-artifact-registry which is not packaged as of 2023-04-25. "tests/pytest_tests/unit_tests_old/tests_launch/test_kaniko_build.py" @@ -255,6 +257,12 @@ buildPythonPackage rec { # Requires metaflow which is not packaged as of 2023-04-25. "tests/pytest_tests/unit_tests/test_metaflow.py" + # Requires tensorflow which is broken as of 2023-09-03 + "tests/pytest_tests/unit_tests/test_keras.py" + + # Try to get hardware information, not possible in the nix build environment + "tests/pytest_tests/unit_tests/test_system_metrics/test_disk.py" + # See https://github.com/wandb/wandb/issues/5423 "tests/pytest_tests/unit_tests/test_docker.py" "tests/pytest_tests/unit_tests/test_library_public.py" |