refact: local_qdm_bc available as a bias resource

Following the pattern, `local_qdm_bc` should be available as a bias resource as well. I missed that when I was implementing it.
NREL · May 22, 2024 · c69951c · c69951c
1 parent 3abc643
commit c69951c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sup3r/bias/__init__.py b/sup3r/bias/__init__.py
@@ -1,14 +1,15 @@
 """Bias calculation and correction modules."""
 
 from .bias_transforms import (global_linear_bc, local_linear_bc,
- monthly_local_linear_bc)
+ local_qdm_bc, monthly_local_linear_bc)
 from .bias_calc import (LinearCorrection, MonthlyLinearCorrection,
  SkillAssessment)
 from .qdm import QuantileDeltaMappingCorrection
 
 __all__ = [
  "global_linear_bc",
  "local_linear_bc",
+ "local_qdm_bc",
  "monthly_local_linear_bc",
  "LinearCorrection",
  "MonthlyLinearCorrection",

diff --git a/tests/bias/test_qdm_bias_correction.py b/tests/bias/test_qdm_bias_correction.py
@@ -11,8 +11,7 @@
 from sup3r import CONFIG_DIR, TEST_DATA_DIR
 from sup3r.models import Sup3rGan
 from sup3r.pipeline.forward_pass import ForwardPass, ForwardPassStrategy
-from sup3r.bias import QuantileDeltaMappingCorrection
-from sup3r.bias.bias_transforms import local_qdm_bc
+from sup3r.bias import local_qdm_bc, QuantileDeltaMappingCorrection
 from sup3r.preprocessing.data_handling import DataHandlerNC, DataHandlerNCforCC
 
 FP_NSRDB = os.path.join(TEST_DATA_DIR, "test_nsrdb_co_2018.h5")