gnu: python-pytorch-lightning: Update to commit 2064887.

* gnu/packages/machine-learning.scm (python-pytorch-lightning): Update to
commit 2064887.
[source]: Update URL.
[arguments]<#:tests-flags>: Skip additional failing tests.
<#:phases>: Adjust 'patch-version-detection' phase.
This commit is contained in:
David Elsing 2024-08-04 22:16:02 +00:00 committed by Ludovic Courtès
parent 77f59d1458
commit 571c605f17
No known key found for this signature in database
GPG key ID: 090B11993D9AEBB5

View file

@ -5157,126 +5157,137 @@ (define-public python-inquirer
(license license:expat)))
(define-public python-pytorch-lightning
(package
(name "python-pytorch-lightning")
(version "2.0.2")
(source (origin
(method git-fetch)
(uri (git-reference
(url "https://github.com/Lightning-AI/lightning")
(commit version)))
(file-name (git-file-name name version))
(sha256
(base32
"1w4lajiql4y5nnhqf6i5wii1mrwnhp5f4bzbwdzb5zz0d0lysb1i"))))
(build-system pyproject-build-system)
(arguments
(list
#:test-flags
'(list "-m" "not cloud and not tpu" "tests/tests_pytorch"
;; we don't have onnxruntime
"--ignore=tests/tests_pytorch/models/test_onnx.py"
(let ((commit "2064887b12dd934a5f9a2bf45897f29e3bfc74d1")
(revision "0"))
(package
(name "python-pytorch-lightning")
(version (git-version "2.3.3" revision commit))
(source (origin
(method git-fetch)
(uri (git-reference
(url "https://github.com/Lightning-AI/pytorch-lightning")
(commit commit)))
(file-name (git-file-name name version))
(sha256
(base32
"1pfmwgzh21i21i4ixank488615q7j8nkvlxd82kmmam97gsd6krg"))))
(build-system pyproject-build-system)
(arguments
(list
#:test-flags
'(list "-m" "not cloud and not tpu" "tests/tests_pytorch"
;; we don't have onnxruntime
"--ignore=tests/tests_pytorch/models/test_onnx.py"
;; We don't have tensorboard, so we skip all those tests that
;; require it for logging.
"--ignore=tests/tests_pytorch/checkpointing/test_model_checkpoint.py"
"--ignore=tests/tests_pytorch/loggers/test_all.py"
"--ignore=tests/tests_pytorch/loggers/test_logger.py"
"--ignore=tests/tests_pytorch/loggers/test_tensorboard.py"
"--ignore=tests/tests_pytorch/models/test_cpu.py"
"--ignore=tests/tests_pytorch/models/test_hparams.py"
"--ignore=tests/tests_pytorch/models/test_restore.py"
"--ignore=tests/tests_pytorch/profilers/test_profiler.py"
"--ignore=tests/tests_pytorch/trainer/flags/test_fast_dev_run.py"
"--ignore=tests/tests_pytorch/trainer/logging_/test_eval_loop_logging.py"
"--ignore=tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py"
"--ignore=tests/tests_pytorch/trainer/properties/test_loggers.py"
"--ignore=tests/tests_pytorch/trainer/properties/test_log_dir.py"
"--ignore=tests/tests_pytorch/trainer/test_trainer.py"
;; We don't have tensorboard, so we skip all those tests that
;; require it for logging.
"--ignore=tests/tests_pytorch/checkpointing/test_model_checkpoint.py"
"--ignore=tests/tests_pytorch/loggers/test_all.py"
"--ignore=tests/tests_pytorch/loggers/test_logger.py"
"--ignore=tests/tests_pytorch/loggers/test_tensorboard.py"
"--ignore=tests/tests_pytorch/loggers/test_wandb.py"
"--ignore=tests/tests_pytorch/models/test_cpu.py"
"--ignore=tests/tests_pytorch/models/test_hparams.py"
"--ignore=tests/tests_pytorch/models/test_restore.py"
"--ignore=tests/tests_pytorch/profilers/test_profiler.py"
"--ignore=tests/tests_pytorch/test_cli.py"
"--ignore=tests/tests_pytorch/trainer/flags/test_fast_dev_run.py"
"--ignore=tests/tests_pytorch/trainer/logging_/test_eval_loop_logging.py"
"--ignore=tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py"
"--ignore=tests/tests_pytorch/trainer/properties/test_loggers.py"
"--ignore=tests/tests_pytorch/trainer/properties/test_log_dir.py"
"--ignore=tests/tests_pytorch/trainer/test_trainer.py"
;; This needs internet access
"--ignore=tests/tests_pytorch/helpers/test_models.py"
"--ignore=tests/tests_pytorch/helpers/test_datasets.py"
"--ignore=tests/tests_pytorch/helpers/datasets.py"
;; This needs internet access
"--ignore=tests/tests_pytorch/helpers/test_models.py"
"--ignore=tests/tests_pytorch/helpers/test_datasets.py"
"--ignore=tests/tests_pytorch/helpers/datasets.py"
;; We have no legacy checkpoints
"--ignore=tests/tests_pytorch/checkpointing/test_legacy_checkpoints.py"
;; We have no legacy checkpoints
"--ignore=tests/tests_pytorch/checkpointing/test_legacy_checkpoints.py"
;; TypeError: _FlakyPlugin._make_test_flaky() got an unexpected keyword argument 'reruns'
"--ignore=tests/tests_pytorch/models/test_amp.py"
"--ignore=tests/tests_pytorch/profilers/test_profiler.py"
;; TypeError: _FlakyPlugin._make_test_flaky() got an unexpected
;; keyword argument 'reruns'
"--ignore=tests/tests_pytorch/models/test_amp.py"
"--ignore=tests/tests_pytorch/utilities/test_all_gather_grad.py"
"--ignore=tests/tests_pytorch/graveyard/test_legacy_import_unpickler.py"
;; Requires CUDA
"--ignore=tests/tests_pytorch/plugins/precision/test_bitsandbytes.py"
"-k"
(string-append
;; We don't have tensorboard
"not test_property_logger"
" and not test_cli_logger_shorthand"
;; Something wrong with Flaky
" and not test_servable_module_validator_with_trainer"))
#:phases
'(modify-phases %standard-phases
(add-after 'unpack 'patch-version-detection
(lambda _
;; We do have pytorch 1.13.1, but the version comparison fails.
(substitute* "src/lightning/fabric/utilities/imports.py"
(("_TORCH_GREATER_EQUAL_1_13 =.*")
"_TORCH_GREATER_EQUAL_1_13 = True\n"))))
(add-before 'build 'pre-build
(lambda _ (setenv "PACKAGE_NAME" "lightning")))
(add-after 'install 'pre-build-pytorch
(lambda _
;; pyproject-build-system only tolerates unicycles.
(for-each delete-file (find-files "dist" "\\.whl"))
(setenv "PACKAGE_NAME" "pytorch")))
(add-after 'pre-build-pytorch 'build-pytorch
(assoc-ref %standard-phases 'build))
(add-after 'build-pytorch 'install-pytorch
(assoc-ref %standard-phases 'install))
(add-before 'check 'pre-check
(lambda _
;; We don't have Tensorboard
(substitute* "tests/tests_pytorch/test_cli.py"
((" TensorBoardLogger\\(\".\"\\)") "")))))))
(propagated-inputs
(list python-arrow
python-beautifulsoup4
python-croniter
python-dateutils
python-deepdiff
python-fastapi-for-pytorch-lightning
python-fsspec
python-inquirer
python-jsonargparse
python-lightning-cloud
python-lightning-utilities
python-numpy
python-packaging
python-pytorch
python-pyyaml
python-starsessions-for-pytorch-lightning
python-torchmetrics
python-torchvision
python-tqdm
python-traitlets
python-typing-extensions))
(native-inputs
(list python-aiohttp
python-cloudpickle
python-coverage
python-flaky
python-pympler
python-pytest
python-psutil
python-requests-mock
python-scikit-learn))
(home-page "https://lightning.ai/")
(synopsis "Deep learning framework to train, deploy, and ship AI products")
(description
"PyTorch Lightning is just organized PyTorch; Lightning disentangles
"-k"
(string-append
;; We don't have tensorboard
"not test_property_logger"
" and not test_cli_logger_shorthand"
;; Wrong module appears in sys.modules
" and not test_patch_legacy_imports_unified"
;; Missing log message
" and not test_should_stop_early_stopping_conditions_met"
" and not test_fit_loop_done_log_messages"
;; Something wrong with Flaky
" and not test_servable_module_validator_with_trainer"))
#:phases
'(modify-phases %standard-phases
(add-after 'unpack 'patch-version-detection
(lambda _
;; We do have pytorch 2.4.0, but the version comparison fails.
(substitute* "src/lightning/fabric/utilities/imports.py"
(("_TORCH_GREATER_EQUAL_2_4 =.*")
"_TORCH_GREATER_EQUAL_2_4 = True\n"))))
(add-before 'build 'pre-build
(lambda _ (setenv "PACKAGE_NAME" "lightning")))
(add-after 'install 'pre-build-pytorch
(lambda _
;; pyproject-build-system only tolerates unicycles.
(for-each delete-file (find-files "dist" "\\.whl"))
(setenv "PACKAGE_NAME" "pytorch")))
(add-after 'pre-build-pytorch 'build-pytorch
(assoc-ref %standard-phases 'build))
(add-after 'build-pytorch 'install-pytorch
(assoc-ref %standard-phases 'install))
(add-before 'check 'pre-check
(lambda _
;; We don't have Tensorboard
(substitute* "tests/tests_pytorch/test_cli.py"
((" TensorBoardLogger\\(\".\"\\)") "")))))))
(propagated-inputs
(list python-arrow
python-beautifulsoup4
python-croniter
python-dateutils
python-deepdiff
python-fastapi-for-pytorch-lightning
python-fsspec
python-inquirer
python-jsonargparse
python-lightning-cloud
python-lightning-utilities
python-numpy
python-packaging
python-pytorch
python-pyyaml
python-starsessions-for-pytorch-lightning
python-torchmetrics
python-torchvision
python-tqdm
python-traitlets
python-typing-extensions))
(native-inputs
(list python-aiohttp
python-cloudpickle
python-coverage
python-flaky
python-pympler
python-pytest
python-psutil
python-requests-mock
python-scikit-learn))
(home-page "https://lightning.ai/")
(synopsis "Deep learning framework to train, deploy, and ship AI products")
(description
"PyTorch Lightning is just organized PyTorch; Lightning disentangles
PyTorch code to decouple the science from the engineering.")
(license license:asl2.0)))
(license license:asl2.0))))
(define-public python-torchmetrics
(package