[CI] Less warnings in CI (costs) (#1349)

pytorch · Jul 3, 2023 · d12afa0 · d12afa0 · github-actions · Jul 3, 2023
1 parent aedad69
commit d12afa0
Show file tree

Hide file tree

Showing 6 changed files with 985 additions and 792 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -4,7 +4,9 @@
 # LICENSE file in the root directory of this source tree.
 
 import os
+
 import time
+import warnings
 from collections import defaultdict
 
 import pytest
@@ -50,3 +52,12 @@ def fin():
         CALL_TIMES[name] = CALL_TIMES[name] + duration
 
     request.addfinalizer(fin)
+
+
+@pytest.fixture(autouse=True)
+def set_warnings() -> None:
+    warnings.filterwarnings(
+        "ignore",
+        category=UserWarning,
+        message=r"Lazy modules are a new feature under heavy development",
+    )
Benchmark suite	Current: `d12afa0`	Previous: `aedad69`	Ratio
`benchmarks/test_objectives_benchmarks.py::test_gae_speed[vec_generalized_advantage_estimate-False-32-512]`	`155.92547707758555` iter/sec (`stddev: 0.0006705226136095314`)	`318.3331871582808` iter/sec (`stddev: 0.00022329479782011772`)	`2.04`
`benchmarks/test_objectives_benchmarks.py::test_redq_deprec_speed`	`35.703702585143205` iter/sec (`stddev: 0.01848743416046796`)	`76.43772886715402` iter/sec (`stddev: 0.000468364400326501`)	`2.14`
`benchmarks/test_objectives_benchmarks.py::test_cql_speed`	`15.331344246775672` iter/sec (`stddev: 0.002779792502851907`)	`33.81095039866202` iter/sec (`stddev: 0.0013921416623026667`)	`2.21`
`benchmarks/test_objectives_benchmarks.py::test_a2c_speed`	`84.88992145328808` iter/sec (`stddev: 0.0011329295705112604`)	`172.1480547376838` iter/sec (`stddev: 0.000761102456951268`)	`2.03`
`benchmarks/test_objectives_benchmarks.py::test_ppo_speed`	`79.88411647441444` iter/sec (`stddev: 0.0011337427643802337`)	`166.43663434536572` iter/sec (`stddev: 0.0003116689495854205`)	`2.08`
`benchmarks/test_objectives_benchmarks.py::test_reinforce_speed`	`100.8847541348576` iter/sec (`stddev: 0.0012163373877399184`)	`223.08673866180482` iter/sec (`stddev: 0.0012326454562100583`)	`2.21`
`benchmarks/test_objectives_benchmarks.py::test_iql_speed`	`20.232033309166678` iter/sec (`stddev: 0.003229574906641596`)	`44.40574481639754` iter/sec (`stddev: 0.0007979074900549904`)	`2.19`