refactor: move remote_function logic to make a space for other types (#322)

Author: tswast

bigframes/functions/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2024 Google LLC
+#
+# 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.

bigframes/remote_function.py renamed to bigframes/functions/remote_function.py

@@ -74,8 +74,8 @@
 import bigframes.core.utils as utils
 import bigframes.dataframe as dataframe
 import bigframes.formatting_helpers as formatting_helpers
-from bigframes.remote_function import read_gbq_function as bigframes_rgf
-from bigframes.remote_function import remote_function as bigframes_rf
+from bigframes.functions.remote_function import read_gbq_function as bigframes_rgf
+from bigframes.functions.remote_function import remote_function as bigframes_rf
 import bigframes.session._io.bigquery as bigframes_io
 import bigframes.session.clients
 import bigframes.version

bigframes/session/__init__.py

@@ -28,7 +28,7 @@
 import test_utils.prefixer
 
 import bigframes
-from bigframes.remote_function import (
+from bigframes.functions.remote_function import (
     get_cloud_function_name,
     get_remote_function_locations,
 )

tests/system/large/test_remote_function.py

@@ -17,7 +17,7 @@
 import pytest
 
 import bigframes
-from bigframes import remote_function as rf
+from bigframes.functions import remote_function as rf
 from tests.system.utils import assert_pandas_df_equal
 
 

tests/system/small/test_remote_function.py

@@ -241,7 +241,7 @@ def test_literal_to_ibis_scalar_throws_on_incompatible_literal():
 def test_remote_function_io_types_are_supported_bigframes_types():
     from ibis.expr.datatypes.core import dtype as python_type_to_bigquery_type
 
-    from bigframes.remote_function import (
+    from bigframes.functions.remote_function import (
         SUPPORTED_IO_PYTHON_TYPES as rf_supported_io_types,
     )
 

tests/unit/test_dtypes.py

@@ -15,7 +15,7 @@
 from ibis.backends.bigquery import datatypes as bq_types
 from ibis.expr import datatypes as ibis_types
 
-from bigframes import remote_function as rf
+from bigframes.functions import remote_function as rf
 
 
 def test_supported_types_correspond():