fix lint errors

Author: tswast

bigframes/bigquery/__init__.py

@@ -18,7 +18,7 @@
 
 import sys
 
-from bigframes.bigquery import ai
+from bigframes.bigquery import ai, ml
 from bigframes.bigquery._operations.approx_agg import approx_top_count
 from bigframes.bigquery._operations.array import (
     array_agg,
@@ -157,4 +157,5 @@
     "struct",
     # Modules / SQL namespaces
     "ai",
+    "ml",
 ]

bigframes/bigquery/_operations/ml.py

@@ -14,7 +14,6 @@
 
 from __future__ import annotations
 
-import typing
 from typing import Mapping, Optional, TYPE_CHECKING, Union
 
 import bigframes.core.log_adapter as log_adapter

bigframes/bigquery/ml.py

@@ -12,7 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""This module integrates BigQuery ML functions."""
+"""This module exposes `BigQuery ML
+<https://docs.cloud.google.com/bigquery/docs/bqml-introduction>`_ functions
+by directly mapping to the equivalent function names in SQL syntax.
+
+For an interface more familiar to Scikit-Learn users, see :mod:`bigframes.ml`.
+"""
 
 from bigframes.bigquery._operations.ml import create_model
 

bigframes/core/sql/ml.py

@@ -14,7 +14,6 @@
 
 from __future__ import annotations
 
-import typing
 from typing import Mapping, Optional, Union
 
 import bigframes.core.compile.googlesql as googlesql

tests/unit/core/sql/test_ml.py

@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import pytest
-
 import bigframes.core.sql.ml
 
 
@@ -25,6 +23,7 @@ def test_create_model_basic(snapshot):
     )
     snapshot.assert_match(sql, "create_model_basic.sql")
 
+
 def test_create_model_replace(snapshot):
     sql = bigframes.core.sql.ml.create_model_ddl(
         model_name="my_model",
@@ -34,6 +33,7 @@ def test_create_model_replace(snapshot):
     )
     snapshot.assert_match(sql, "create_model_replace.sql")
 
+
 def test_create_model_if_not_exists(snapshot):
     sql = bigframes.core.sql.ml.create_model_ddl(
         model_name="my_model",
@@ -43,6 +43,7 @@ def test_create_model_if_not_exists(snapshot):
     )
     snapshot.assert_match(sql, "create_model_if_not_exists.sql")
 
+
 def test_create_model_transform(snapshot):
     sql = bigframes.core.sql.ml.create_model_ddl(
         model_name="my_model",
@@ -52,6 +53,7 @@ def test_create_model_transform(snapshot):
     )
     snapshot.assert_match(sql, "create_model_transform.sql")
 
+
 def test_create_model_remote(snapshot):
     sql = bigframes.core.sql.ml.create_model_ddl(
         model_name="my_remote_model",
@@ -62,6 +64,7 @@ def test_create_model_remote(snapshot):
     )
     snapshot.assert_match(sql, "create_model_remote.sql")
 
+
 def test_create_model_remote_default(snapshot):
     sql = bigframes.core.sql.ml.create_model_ddl(
         model_name="my_remote_model",
@@ -70,6 +73,7 @@ def test_create_model_remote_default(snapshot):
     )
     snapshot.assert_match(sql, "create_model_remote_default.sql")
 
+
 def test_create_model_training_data_and_holiday(snapshot):
     sql = bigframes.core.sql.ml.create_model_ddl(
         model_name="my_arima_model",
@@ -79,6 +83,7 @@ def test_create_model_training_data_and_holiday(snapshot):
     )
     snapshot.assert_match(sql, "create_model_training_data_and_holiday.sql")
 
+
 def test_create_model_list_option(snapshot):
     sql = bigframes.core.sql.ml.create_model_ddl(
         model_name="my_model",