dfm-python 0.5.4__py3-none-any.whl → 0.5.5__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- dfm_python/__init__.py +5 -1
- dfm_python/config/__init__.py +4 -2
- dfm_python/config/adapter.py +474 -131
- dfm_python/config/results.py +6 -3
- dfm_python/config/schema.py +319 -433
- dfm_python/config/utils.py +9 -21
- dfm_python/data/utils.py +10 -12
- dfm_python/decoder/__init__.py +0 -0
- dfm_python/lightning/__init__.py +6 -6
- dfm_python/lightning/ddfm_dm.py +581 -0
- dfm_python/lightning/dfm_dm.py +379 -0
- dfm_python/lightning/utils.py +496 -0
- dfm_python/models/base.py +146 -191
- dfm_python/models/ddfm.py +84 -69
- dfm_python/models/dfm.py +362 -311
- dfm_python/models/mcmc.py +9 -15
- dfm_python/ssm/em.py +581 -910
- dfm_python/ssm/kalman.py +265 -468
- dfm_python/ssm/utils.py +139 -166
- dfm_python/trainer/ddfm.py +2 -2
- {dfm_python-0.5.4.dist-info → dfm_python-0.5.5.dist-info}/METADATA +1 -1
- dfm_python-0.5.5.dist-info/RECORD +46 -0
- dfm_python/lightning/data_module.py +0 -771
- dfm_python/lightning/scaling.py +0 -444
- dfm_python-0.5.4.dist-info/RECORD +0 -44
- {dfm_python-0.5.4.dist-info → dfm_python-0.5.5.dist-info}/WHEEL +0 -0
- {dfm_python-0.5.4.dist-info → dfm_python-0.5.5.dist-info}/licenses/LICENSE +0 -0
dfm_python/__init__.py
CHANGED
|
@@ -65,7 +65,7 @@ Note: DFMConfig and SeriesConfig are internal implementation details.
|
|
|
65
65
|
For detailed documentation, see the README.md file and the tutorial notebooks/scripts.
|
|
66
66
|
"""
|
|
67
67
|
|
|
68
|
-
__version__ = "0.5.
|
|
68
|
+
__version__ = "0.5.5"
|
|
69
69
|
|
|
70
70
|
# ============================================================================
|
|
71
71
|
# PUBLIC API DEFINITION
|
|
@@ -102,7 +102,9 @@ from .utils.time import calculate_rmse
|
|
|
102
102
|
# Users can import these directly from dfm_python
|
|
103
103
|
from .lightning import (
|
|
104
104
|
DFMDataModule,
|
|
105
|
+
DDFMDataModule, # DDFM-specific DataModule
|
|
105
106
|
DFMDataset, # Dataset class (usually not needed directly)
|
|
107
|
+
DDFMDataset, # DDFM Dataset class (usually not needed directly)
|
|
106
108
|
KalmanFilter, # Module class
|
|
107
109
|
EMAlgorithm, # Module class
|
|
108
110
|
)
|
|
@@ -137,7 +139,9 @@ __all__.extend([
|
|
|
137
139
|
# Lightning modules (mandatory dependency)
|
|
138
140
|
__all__.extend([
|
|
139
141
|
'DFMDataModule',
|
|
142
|
+
'DDFMDataModule', # DDFM-specific DataModule
|
|
140
143
|
'DFMDataset', # Dataset class (usually not needed directly)
|
|
144
|
+
'DDFMDataset', # DDFM Dataset class (usually not needed directly)
|
|
141
145
|
'KalmanFilter', # Module class
|
|
142
146
|
'EMAlgorithm', # Module class
|
|
143
147
|
])
|
dfm_python/config/__init__.py
CHANGED
|
@@ -14,7 +14,9 @@ from .utils import validate_frequency, validate_transformation
|
|
|
14
14
|
from .adapter import (
|
|
15
15
|
ConfigSource,
|
|
16
16
|
YamlSource,
|
|
17
|
+
DictSource,
|
|
17
18
|
HydraSource,
|
|
19
|
+
MergedConfigSource,
|
|
18
20
|
make_config_source,
|
|
19
21
|
)
|
|
20
22
|
from .utils import (
|
|
@@ -41,8 +43,8 @@ __all__ = [
|
|
|
41
43
|
# Utilities
|
|
42
44
|
'validate_frequency', 'validate_transformation',
|
|
43
45
|
# IO
|
|
44
|
-
'ConfigSource', 'YamlSource',
|
|
45
|
-
'HydraSource', 'make_config_source',
|
|
46
|
+
'ConfigSource', 'YamlSource', 'DictSource',
|
|
47
|
+
'HydraSource', 'MergedConfigSource', 'make_config_source',
|
|
46
48
|
# Frequency and aggregation utilities
|
|
47
49
|
'FREQUENCY_HIERARCHY',
|
|
48
50
|
'PERIODS_PER_YEAR',
|