Update (base update)

[ghstack-poisoned]
pytorch · Feb 5, 2025 · 3ca1767 · 3ca1767
2 parents 8267e94 + bae04ce
commit 3ca1767
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/.github/scripts/linux-post-script.sh 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
@@ -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
@@ -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