Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Reorganizing QDM related methods #216

Merged
merged 8 commits into from
May 24, 2024
Merged
Prev Previous commit
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.
  • Loading branch information
castelao committed May 22, 2024
commit c69951c60fe614780c265b6643ffde41b8171aa3
3 changes: 2 additions & 1 deletion sup3r/bias/__init__.py
Original file line number Diff line number Diff line change
@@ -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",
Expand Down
3 changes: 1 addition & 2 deletions tests/bias/test_qdm_bias_correction.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down
Loading