Skip to content

Commit 53e16ba

Browse files
yyyu-googlecopybara-github
authored andcommitted
feat: migrate batch_predictions to agentplatform
PiperOrigin-RevId: 904559578
1 parent 3d380ba commit 53e16ba

9 files changed

Lines changed: 1680 additions & 20 deletions

File tree

agentplatform/__init__.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# Copyright 2026 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
#
15+
"""The agentplatform module."""
16+
17+
from google.cloud.aiplatform import init
18+
from google.cloud.aiplatform import version as aiplatform_version
19+
20+
__version__ = aiplatform_version.__version__
21+
22+
__all__ = [
23+
"init",
24+
]
Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
# Copyright 2026 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
#
15+
"""Classes for batch prediction."""
16+
17+
# We just want to re-export certain classes
18+
# pylint: disable=g-multiple-import,g-importing-member
19+
from agentplatform.batch_prediction._batch_prediction import (
20+
BatchPredictionJob,
21+
)
22+
23+
__all__ = [
24+
"BatchPredictionJob",
25+
]

0 commit comments

Comments
 (0)