Skip to content

Commit 88c2cac

Browse files
santoso-wijayaThe Meridian Authors
santoso-wijaya
authored and
The Meridian Authors
committed
Code update
PiperOrigin-RevId: 868368199
1 parent ab5bcf0 commit 88c2cac

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

52 files changed

+118
-118
lines changed

schema/__init__.py renamed to meridian/schema/__init__.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
) from exc
2727

2828
# pylint: disable=g-import-not-at-top
29-
from schema import mmm_proto_generator
30-
from schema import model_consumer
31-
from schema import processors
32-
from schema import serde
33-
from schema import utils
29+
from meridian.schema import mmm_proto_generator
30+
from meridian.schema import model_consumer
31+
from meridian.schema import processors
32+
from meridian.schema import serde
33+
from meridian.schema import utils

schema/mmm_proto_generator.py renamed to meridian/schema/mmm_proto_generator.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,12 @@
2323

2424
from meridian.model import model
2525
from mmm.v1 import mmm_pb2 as mmm_pb
26-
from schema import model_consumer
27-
from schema.processors import budget_optimization_processor
28-
from schema.processors import marketing_processor
29-
from schema.processors import model_fit_processor
30-
from schema.processors import model_processor
31-
from schema.processors import reach_frequency_optimization_processor
26+
from meridian.schema import model_consumer
27+
from meridian.schema.processors import budget_optimization_processor
28+
from meridian.schema.processors import marketing_processor
29+
from meridian.schema.processors import model_fit_processor
30+
from meridian.schema.processors import model_processor
31+
from meridian.schema.processors import reach_frequency_optimization_processor
3232

3333

3434
__all__ = [

schema/mmm_proto_generator_test.py renamed to meridian/schema/mmm_proto_generator_test.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -18,14 +18,14 @@
1818
from absl.testing import absltest
1919
from meridian.analysis import optimizer
2020
from mmm.v1.model import mmm_kernel_pb2 as kernel_pb
21-
from schema import mmm_proto_generator
22-
from schema import test_data as td
23-
from schema.processors import budget_optimization_processor
24-
from schema.processors import marketing_processor
25-
from schema.processors import model_fit_processor
26-
from schema.processors import model_processor
27-
from schema.processors import reach_frequency_optimization_processor as rf_opt_processor
28-
from schema.serde import meridian_serde
21+
from meridian.schema import mmm_proto_generator
22+
from meridian.schema import test_data as td
23+
from meridian.schema.processors import budget_optimization_processor
24+
from meridian.schema.processors import marketing_processor
25+
from meridian.schema.processors import model_fit_processor
26+
from meridian.schema.processors import model_processor
27+
from meridian.schema.processors import reach_frequency_optimization_processor as rf_opt_processor
28+
from meridian.schema.serde import meridian_serde
2929

3030
from tensorflow.python.util.protobuf import compare
3131

schema/model_consumer.py renamed to meridian/schema/model_consumer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@
2525

2626
from meridian.model import model
2727
from mmm.v1 import mmm_pb2 as mmm_pb
28-
from schema.processors import model_kernel_processor
29-
from schema.processors import model_processor
28+
from meridian.schema.processors import model_kernel_processor
29+
from meridian.schema.processors import model_processor
3030

3131

3232
__all__ = [

schema/model_consumer_test.py renamed to meridian/schema/model_consumer_test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@
2121
from mmm.v1.fit import model_fit_pb2 as fit_pb
2222
from mmm.v1.marketing.optimization import marketing_optimization_pb2 as opt_pb
2323
from mmm.v1.model import mmm_kernel_pb2 as kernel_pb
24-
from schema import model_consumer
25-
from schema.processors import model_processor
26-
from schema.serde import meridian_serde
24+
from meridian.schema import model_consumer
25+
from meridian.schema.processors import model_processor
26+
from meridian.schema.serde import meridian_serde
2727

2828

2929
class FooSpec(model_processor.Spec):

schema/processors/__init__.py renamed to meridian/schema/processors/__init__.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -68,10 +68,10 @@
6868
of the individual modules.
6969
"""
7070

71-
from schema.processors import budget_optimization_processor
72-
from schema.processors import common
73-
from schema.processors import marketing_processor
74-
from schema.processors import model_fit_processor
75-
from schema.processors import model_kernel_processor
76-
from schema.processors import model_processor
77-
from schema.processors import reach_frequency_optimization_processor
71+
from meridian.schema.processors import budget_optimization_processor
72+
from meridian.schema.processors import common
73+
from meridian.schema.processors import marketing_processor
74+
from meridian.schema.processors import model_fit_processor
75+
from meridian.schema.processors import model_kernel_processor
76+
from meridian.schema.processors import model_processor
77+
from meridian.schema.processors import reach_frequency_optimization_processor

schema/processors/budget_optimization_processor.py renamed to meridian/schema/processors/budget_optimization_processor.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,9 @@
4545
Optimize budget allocation for a fixed total budget, aiming to maximize KPI.
4646
4747
```python
48-
from schema.processors import budget_optimization_processor
48+
from meridian.schema.processors import budget_optimization_processor
4949
from meridian.analysis import optimizer
50-
from schema.processors import common
50+
from meridian.schema.processors import common
5151
5252
# Assuming 'trained_model' is a loaded Meridian model object
5353
@@ -82,9 +82,9 @@
8282
Optimize budget to achieve a target Return on Investment (ROI).
8383
8484
```python
85-
from schema.processors import budget_optimization_processor
85+
from meridian.schema.processors import budget_optimization_processor
8686
from meridian.analysis import optimizer
87-
from schema.processors import common
87+
from meridian.schema.processors import common
8888
import meridian.constants as c
8989
9090
# Assuming 'trained_model' is a loaded Meridian model object
@@ -130,9 +130,9 @@
130130
from mmm.v1.marketing.analysis import response_curve_pb2 as response_curve_pb
131131
from mmm.v1.marketing.optimization import budget_optimization_pb2 as budget_pb
132132
from mmm.v1.marketing.optimization import constraints_pb2 as constraints_pb
133-
from schema.processors import common
134-
from schema.processors import model_processor
135-
from schema.utils import time_record
133+
from meridian.schema.processors import common
134+
from meridian.schema.processors import model_processor
135+
from meridian.schema.utils import time_record
136136
import numpy as np
137137
from typing_extensions import override
138138
import xarray as xr

schema/processors/common.py renamed to meridian/schema/processors/common.py

File renamed without changes.

schema/processors/common_test.py renamed to meridian/schema/processors/common_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from absl.testing import absltest
1616
from meridian import constants
1717
from mmm.v1.common import estimate_pb2 as estimate_pb
18-
from schema.processors import common
18+
from meridian.schema.processors import common
1919
import numpy as np
2020
import xarray as xr
2121

schema/processors/marketing_processor.py renamed to meridian/schema/processors/marketing_processor.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
1. **Get Media Summary Metrics for a specific period:**
5050
5151
```python
52-
from schema.processors import marketing_processor
52+
from meridian.schema.processors import marketing_processor
5353
import datetime
5454
5555
# Assuming 'trained_model' is a loaded Meridian model object
@@ -73,7 +73,7 @@
7373
2. **Calculate Incremental Outcome with new spend data:**
7474
7575
```python
76-
from schema.processors import marketing_processor
76+
from meridian.schema.processors import marketing_processor
7777
from meridian.analysis import analyzer
7878
import datetime
7979
import numpy as np
@@ -127,8 +127,8 @@
127127
from mmm.v1.marketing.analysis import non_media_analysis_pb2
128128
from mmm.v1.marketing.analysis import outcome_pb2
129129
from mmm.v1.marketing.analysis import response_curve_pb2
130-
from schema.processors import common
131-
from schema.processors import model_processor
130+
from meridian.schema.processors import common
131+
from meridian.schema.processors import model_processor
132132
import numpy as np
133133
import xarray as xr
134134

0 commit comments

Comments
 (0)