Add tracing_lambdas attribute to benchmark class

tardis-sn · Aug 9, 2023 · 84c257f · 84c257f
1 parent d518658
commit 84c257f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml
@@ -6,7 +6,7 @@ env:
 on:
   push:
     branches:
-      - "*"
+      - main
   pull_request_target:
     branches:
       - "*"

diff --git a/benchmarks/run_stardis.py b/benchmarks/run_stardis.py
@@ -3,7 +3,6 @@
 import os
 import numpy as np
 from stardis.base import run_stardis
-
 from astropy import units as u
 
 from tardis.io.atom_data import AtomData
@@ -58,6 +57,7 @@ def setup(self):
         self.stellar_model = stellar_model
         self.alphas = alphas
         self.tracing_nus = tracing_nus
+        self.tracing_lambdas = tracing_lambdas
         self.config = config
 
     def time_run_stardis(self):
@@ -77,4 +77,4 @@ def time_calc_alpha(self):
             self.stellar_model,
             self.tracing_nus,
             self.config.opacity.line,
-        )
+        )