Skip to content

Commit 0da806b

Browse files
committed
Use WavelengthBins instead of separate parameters for min, max and number of bins
1 parent 39b9ed1 commit 0da806b

2 files changed

Lines changed: 8 additions & 10 deletions

File tree

docs/examples/scaling_workflow.ipynb

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -73,13 +73,11 @@
7373
"import sciline as sl\n",
7474
"import scipp as sc\n",
7575
"\n",
76-
"from ess.nmx.mtz_io import (\n",
77-
" providers as mtz_io_providers,\n",
78-
" default_parameters as mtz_io_params,\n",
79-
")\n",
76+
"from ess.nmx.mtz_io import providers as mtz_io_providers, default_parameters as mtz_io_params\n",
8077
"from ess.nmx.mtz_io import SpaceGroupDesc\n",
81-
"from ess.nmx.scaling import scaling_providers, scaling_params\n",
78+
"from ess.nmx.scaling import providers as scaling_providers, default_parameters as scaling_params\n",
8279
"from ess.nmx.scaling import (\n",
80+
" WavelengthBins,\n",
8381
" FilteredEstimatedScaledIntensities,\n",
8482
" ReferenceWavelength,\n",
8583
" ScaledIntensityLeftTailThreshold,\n",
@@ -101,6 +99,7 @@
10199
" ),\n",
102100
" **mtz_io_params,\n",
103101
" **scaling_params,\n",
102+
" WavelengthBins: 250,\n",
104103
" },\n",
105104
")\n",
106105
"pl"

src/ess/nmx/scaling.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,14 @@
22
# Copyright (c) 2024 Scipp contributors (https://github.com/scipp)
33
from collections.abc import Callable, Mapping
44
from dataclasses import dataclass
5-
from typing import NewType, TypeVar
5+
from typing import NewType, TypeVar, Union
66

77
import scipp as sc
88

99
from .mtz_io import NMXMtzDataArray
1010

1111
# User defined or configurable types
12-
WavelengthBins = NewType("WavelengthBins", sc.Variable)
12+
WavelengthBins = NewType("WavelengthBins", Union[sc.Variable, int])
1313
"""User configurable wavelength binning"""
1414
ReferenceWavelength = NewType("ReferenceWavelength", sc.Variable | None)
1515
"""The wavelength to select reference intensities."""
@@ -438,8 +438,7 @@ def calculate_wavelength_scale_factor(
438438
return WavelengthScaleFactors(scale_factor)
439439

440440

441-
# Providers and default parameters
442-
scaling_providers = (
441+
providers = (
443442
cut_tails,
444443
get_wavelength_binned,
445444
get_reference_wavelength,
@@ -451,7 +450,7 @@ def calculate_wavelength_scale_factor(
451450
)
452451
"""Providers for scaling data."""
453452

454-
scaling_params = {
453+
default_parameters = {
455454
WavelengthBins: sc.linspace("wavelength", 2.6, 3.6, 250, unit="angstrom"),
456455
ScaledIntensityLeftTailThreshold: DEFAULT_LEFT_TAIL_THRESHOLD,
457456
ScaledIntensityRightTailThreshold: DEFAULT_RIGHT_TAIL_THRESHOLD,

0 commit comments

Comments
 (0)