From bae04cedd9e6877efd9dbf00771a8ac4d7108697 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Wed, 5 Feb 2025 11:27:54 +0000 Subject: [PATCH] [CI] Fix GLIBCXX_3.4.29 error (#1207) --- .github/scripts/linux-post-script.sh | 4 ++++ .github/workflows/build-wheels-linux.yml | 3 ++- tensordict/nn/distributions/composite.py | 2 +- 3 files changed, 7 insertions(+), 2 deletions(-) create mode 100755 .github/scripts/linux-post-script.sh diff --git a/.github/scripts/linux-post-script.sh b/.github/scripts/linux-post-script.sh new file mode 100755 index 000000000..537989045 --- /dev/null +++ b/.github/scripts/linux-post-script.sh @@ -0,0 +1,4 @@ +#!/bin/bash + +yum update gcc +yum update libstdc++ diff --git a/.github/workflows/build-wheels-linux.yml b/.github/workflows/build-wheels-linux.yml index 09afd984d..8bf29802c 100644 --- a/.github/workflows/build-wheels-linux.yml +++ b/.github/workflows/build-wheels-linux.yml @@ -33,7 +33,7 @@ jobs: include: - repository: pytorch/tensordict smoke-test-script: test/smoke_test.py - env-script: .github/scripts/install-deps-smoke-test.sh + post-script: .github/scripts/linux-post-script.sh package-name: tensordict name: pytorch/tensordict uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main @@ -47,3 +47,4 @@ jobs: smoke-test-script: ${{ matrix.smoke-test-script }} trigger-event: ${{ github.event_name }} env-var-script: .github/scripts/version_script.sh + post-script: ${{ matrix.post-script }} diff --git a/tensordict/nn/distributions/composite.py b/tensordict/nn/distributions/composite.py index 32b6d6ed3..1eeb7e990 100644 --- a/tensordict/nn/distributions/composite.py +++ b/tensordict/nn/distributions/composite.py @@ -130,7 +130,7 @@ def __init__( dist_params = params.get(name) kwargs = extra_kwargs.get(name, {}) if dist_params is None: - raise KeyError + raise KeyError(f"no param {name} found in params with keys {params.keys(True, True)}") dist = dist_class(**dist_params, **kwargs) dists[write_name] = dist self.dists = dists