Merge branch 'main' into sycai_ai_gen_text

Author: sycai

tests/system/large/bigquery/test_ml.py

@@ -62,3 +62,30 @@ def test_generate_embedding_with_options(embedding_model):
     assert "ml_generate_embedding_status" in result.columns
     embedding = result["ml_generate_embedding_result"].to_pandas()
     assert len(embedding[0]) == 256
+
+
+def test_create_model_linear_regression(dataset_id):
+    df = bpd.DataFrame({"x": [1, 2, 3], "y": [2, 4, 6]})
+    model_name = f"{dataset_id}.linear_regression_model"
+
+    result = ml.create_model(
+        model_name=model_name,
+        options={"model_type": "LINEAR_REG", "input_label_cols": ["y"]},
+        training_data=df,
+    )
+
+    assert result["modelType"] == "LINEAR_REGRESSION"
+
+
+def test_create_model_with_transform(dataset_id):
+    df = bpd.DataFrame({"x": [1, 2, 3], "y": [2, 4, 6]})
+    model_name = f"{dataset_id}.transform_model"
+
+    result = ml.create_model(
+        model_name=model_name,
+        options={"model_type": "LINEAR_REG", "input_label_cols": ["y"]},
+        training_data=df,
+        transform=["x * 2 AS x_doubled", "y"],
+    )
+
+    assert result["modelType"] == "LINEAR_REGRESSION"

tests/system/large/blob/test_function.py

@@ -26,6 +26,8 @@
 from bigframes import dtypes
 import bigframes.pandas as bpd
 
+pytest.skip("Skipping blob tests due to b/481790217", allow_module_level=True)
+
 
 @pytest.fixture(scope="function")
 def images_output_folder() -> Generator[str, None, None]:

tests/system/small/blob/test_io.py

@@ -20,6 +20,9 @@
 import bigframes
 import bigframes.pandas as bpd
 
+pytest.skip("Skipping blob tests due to b/481790217", allow_module_level=True)
+
+
 idisplay = pytest.importorskip("IPython.display")
 
 

tests/system/small/blob/test_properties.py

@@ -13,10 +13,13 @@
 # limitations under the License.
 
 import pandas as pd
+import pytest
 
 import bigframes.dtypes as dtypes
 import bigframes.pandas as bpd
 
+pytest.skip("Skipping blob tests due to b/481790217", allow_module_level=True)
+
 
 def test_blob_uri(images_uris: list[str], images_mm_df: bpd.DataFrame):
     actual = images_mm_df["blob_col"].blob.uri().to_pandas()

tests/system/small/blob/test_urls.py

@@ -12,8 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import pytest
+
 import bigframes.pandas as bpd
 
+pytest.skip("Skipping blob tests due to b/481790217", allow_module_level=True)
+
 
 def test_blob_read_url(images_mm_df: bpd.DataFrame):
     urls = images_mm_df["blob_col"].blob.read_url()