Skip to content

Commit

Permalink
[Test] Ignore annoying jit_utils warning with device context manager
Browse files Browse the repository at this point in the history
ghstack-source-id: 23fffb80e79bb839b34178cf5e20faea7a8115c5
Pull Request resolved: #1151
  • Loading branch information
vmoens committed Dec 19, 2024
1 parent 42ad066 commit c671d8c
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions test/test_tensordict.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,6 +133,9 @@
pytest.mark.filterwarnings(
"ignore:The content of the stacked NonTensorData objects matched in value but not identity"
),
pytest.mark.filterwarnings(
"ignore:No PYTORCH_KERNEL_CACHE_PATH or HOME environment variable set"
),
]

mp_ctx = "spawn"
Expand Down

0 comments on commit c671d8c

Please sign in to comment.