Skip to content

Commit

Permalink
more CI debugging for pytorch test.
Browse files Browse the repository at this point in the history
  • Loading branch information
peterg17 committed Dec 11, 2024
1 parent 30d9755 commit 471c6f1
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions tests/profiling/test_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,10 +39,6 @@ def test_call_script_gevent(monkeypatch):

@pytest.mark.skipif(not os.getenv("DD_PROFILING_PYTORCH_ENABLED", False), reason="Not testing pytorch GPU")
def test_call_script_pytorch_gpu(monkeypatch):
monkeypatch.setenv("DD_PROFILING_API_TIMEOUT", "0.1")
monkeypatch.setenv("DD_PROFILING_ENABLED", "true")
monkeypatch.setenv("DD_PROFILING_PYTORCH_ENABLED", "true")
monkeypatch.setenv("DD_PROFILING_EXPORT_LIBDD_ENABLED", "true")
stdout, stderr, exitcode, pid = call_program(
"ddtrace-run", sys.executable, os.path.join(os.path.dirname(__file__), "simple_program_pytorch_gpu.py")
)
Expand Down

0 comments on commit 471c6f1

Please sign in to comment.