actually push tests

ADA-research · Oct 13, 2023 · b66e6cf · b66e6cf
1 parent b338ae2
commit b66e6cf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/test_portfolio/test_portfolio_runner.py b/tests/test_portfolio/test_portfolio_runner.py
@@ -20,9 +20,9 @@ def faulty_pf(pf_cfg: Configuration, pf_cfg2: Configuration) -> Portfolio:
 
 def test_init_runner(portfolio: Portfolio, faulty_pf: Portfolio):
     with pytest.raises(RuntimeError):
-        _ = PortfolioRunner(faulty_pf)
+        _ = PortfolioRunner(faulty_pf, n_cpu=16, n_gpu=1)
 
-    runner = PortfolioRunner(portfolio)
+    runner = PortfolioRunner(portfolio, n_cpu=16, n_gpu=1)
     allocs = [(8, 15, -1), (0, 7, 0)]
     allocs2 = [(8, 15, 0), (0, 7, -1)]
 

diff --git a/tests/test_portfolio/test_resource_tracker.py b/tests/test_portfolio/test_resource_tracker.py
@@ -28,7 +28,9 @@ def resource_tracker(
         alpha=1,
     )
 
-    rt = ResourceTracker(pf_scen, strategy=ResourceStrategy.Auto)
+    rt = ResourceTracker(
+        pf_scen, strategy=ResourceStrategy.Auto, cpu_gpu_count=(16, 1)
+    )
     return rt