diff --git a/.github/workflows/linting.yml b/.github/workflows/linting.yml index 6098a6e..c253599 100644 --- a/.github/workflows/linting.yml +++ b/.github/workflows/linting.yml @@ -23,6 +23,7 @@ jobs: sudo apt-get update python -m pip install --upgrade pip sudo apt install libbz2-dev + conda install -c conda-forge somoclu pip install wheel pip install . pip install .[dev] diff --git a/.github/workflows/testing-and-coverage.yml b/.github/workflows/testing-and-coverage.yml index 032b2cd..67c09f4 100644 --- a/.github/workflows/testing-and-coverage.yml +++ b/.github/workflows/testing-and-coverage.yml @@ -28,6 +28,7 @@ jobs: sudo apt-get update python -m pip install --upgrade pip sudo apt install libbz2-dev + conda install -c conda-forge somoclu pip install wheel pip install . pip install .[dev] diff --git a/src/rail/pipelines/estimation/tomography.py b/src/rail/pipelines/estimation/tomography.py index 927adf8..f2419fd 100644 --- a/src/rail/pipelines/estimation/tomography.py +++ b/src/rail/pipelines/estimation/tomography.py @@ -22,6 +22,7 @@ def __init__( classifiers: dict | None=None, summarizers: dict | None=None, n_tomo_bins: int=5, + nsamples: int=20 ): RailPipeline.__init__(self) @@ -53,7 +54,7 @@ def __init__( self.default_input_dict[f"input_{pz_algo_name_}"] = 'dummy.in' self.add_stage(the_classifier) - for ibin in range(n_tomo_bins): + for ibin in range(1, n_tomo_bins+1): true_nz = TrueNZHistogrammer.make_and_connect( name=f"true_nz_{pz_algo_name_}_{classifier_name_}_bin{ibin}", @@ -77,6 +78,6 @@ def __init__( tomography_bins=the_classifier.io.output, ), selected_bin=ibin, - nsamples=20, + nsamples=nsamples, ) self.add_stage(the_summarizer)