Move schema package as meridian.schema

PiperOrigin-RevId: 868851002

Author: santoso-wijaya

.github/workflows/ci.yml

@@ -94,30 +94,6 @@ jobs:
       # Run tests (in parallel)
       - name: Run core tests
         if: ${{ !contains(github.event.head_commit.message, '#skip-pytest') }}
-        run: pytest -vv --ignore=schema
+        run: pytest -vv
         env:
           MERIDIAN_BACKEND: ${{ matrix.meridian-backend }}
-
-  pytest-schema:
-    runs-on: ubuntu-latest
-    timeout-minutes: 20
-    name: pytest-schema
-    steps:
-      - uses: actions/checkout@v4
-        if: ${{ !contains(github.event.head_commit.message, '#skip-pytest') }}
-      - uses: actions/setup-python@v5
-        if: ${{ !contains(github.event.head_commit.message, '#skip-pytest') }}
-        with:
-          python-version: ${{ matrix.python-version }}
-          cache: pip
-          cache-dependency-path: '**/pyproject.toml'
-      - name: Install proto dependencies and test tools
-        if: ${{ !contains(github.event.head_commit.message, '#skip-pytest') }}
-        run: |
-          pip install semver
-          pip install -e proto --config-settings editable_mode=strict
-          pip install -e . --config-settings editable_mode=strict
-          pip install pytest pytest-xdist
-      - name: Run schema tests
-        if: ${{ !contains(github.event.head_commit.message, '#skip-pytest') }}
-        run: pytest -vv -n auto schema

CHANGELOG.md

@@ -23,6 +23,10 @@ To release a new version (e.g. from `1.0.0` -> `2.0.0`):
 
 ## [Unreleased]
 
+* Move `schema` package as `meridian.schema`.
+  * Introduce a temporary shim `schema.py` module for backwards compatibility.
+    This shim will be removed in the next cut release.
+
 ## [1.5.1] - 2026-02-04
 
 * Fix serialization for binary and text files.

meridian/__init__.py

@@ -13,13 +13,19 @@
 # limitations under the License.
 
 """Meridian API."""
+
 from meridian import analysis
 from meridian import backend
 from meridian import data
 from meridian import model
-from meridian.version import __version__
+from meridian.version import __version__  # pylint: disable=g-importing-member
 
 try:
   from meridian import mlflow  # pylint: disable=g-import-not-at-top
 except ImportError:
   pass
+
+try:
+  from meridian import schema
+except ImportError:
+  pass

pyproject.toml

@@ -110,9 +110,10 @@ build-backend = "setuptools.build_meta"
 
 [tool.setuptools]
 include-package-data = true
+py-modules = ["schema"]  # TODO: Remove this shim.
 
 [tool.setuptools.packages.find]
-include = ["meridian*", "schema*", "scenarioplanner*"]
+include = ["meridian*", "scenarioplanner*"]
 exclude = ["*test"]
 
 [tool.setuptools.dynamic]

schema.py

@@ -0,0 +1,30 @@
+# Copyright 2026 The Meridian Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""A temporary shim for backwards compatibility."""
+# TODO: Remove this file in a future release cut.
+
+import warnings
+
+warnings.warn(
+    "This top-level 'schema' import is deprecated and will be removed in a"
+    " future version. Please update your imports to use 'from meridian.schema"
+    " import ...'.",
+    DeprecationWarning,
+    stacklevel=2,
+)
+
+# Import and re-export the contents from the new location
+# pylint: disable=g-import-not-at-top,wildcard-import
+from meridian.schema import *