From 4b14b3aabea8a954a0461b4ed0011e87b30e6d90 Mon Sep 17 00:00:00 2001 From: Aarya Chaumal Date: Wed, 9 Aug 2023 23:44:10 +0530 Subject: [PATCH] Add tracing_lambdas attribute to benchmark class --- .github/workflows/benchmarks.yml | 2 +- benchmarks/run_stardis.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/benchmarks.yml b/.github/workflows/benchmarks.yml index c5c2bd72..6ceff42e 100644 --- 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 index f70012c3..55ff3041 100644 --- 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, - ) \ No newline at end of file + )