sqlframe 2.0.0__py3-none-any.whl → 2.1.0__py3-none-any.whl

sqlframe/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.0.0'
-__version_tuple__ = version_tuple = (2, 0, 0)
+__version__ = version = '2.1.0'
+__version_tuple__ = version_tuple = (2, 1, 0)

sqlframe/base/_typing.py

@@ -24,6 +24,7 @@ OutputExpressionContainer = t.Union[exp.Select, exp.Create, exp.Insert]
 StorageLevel = str
 PathOrPaths = t.Union[str, t.List[str]]
 OptionalPrimitiveType = t.Optional[PrimitiveType]
+DataTypeOrString = t.Union[DataType, str]
 
 
 class UserDefinedFunctionLike(t.Protocol):

sqlframe/base/functions.py

@@ -2070,14 +2070,19 @@ def contains(left: ColumnOrName, right: ColumnOrName) -> Column:
     return Column.invoke_anonymous_function(left, "contains", right)
 
 
-@meta(unsupported_engines="*")
+@meta(unsupported_engines=["bigquery", "postgres"])
 def convert_timezone(
     sourceTz: t.Optional[Column], targetTz: Column, sourceTs: ColumnOrName
 ) -> Column:
-    if sourceTz is None:
-        return Column.invoke_anonymous_function(targetTz, "convert_timezone", sourceTs)
-    else:
-        return Column.invoke_anonymous_function(sourceTz, "convert_timezone", targetTz, sourceTs)
+    to_timestamp = get_func_from_session("to_timestamp")
+
+    return Column(
+        expression.ConvertTimezone(
+            timestamp=to_timestamp(Column.ensure_col(sourceTs)).expression,
+            source_tz=sourceTz.expression if sourceTz else None,
+            target_tz=Column.ensure_col(targetTz).expression,
+        )
+    )
 
 
 @meta(unsupported_engines="postgres")

sqlframe/base/session.py

@@ -27,6 +27,7 @@ from sqlframe.base.catalog import _BaseCatalog
 from sqlframe.base.dataframe import _BaseDataFrame
 from sqlframe.base.normalize import normalize_dict
 from sqlframe.base.readerwriter import _BaseDataFrameReader, _BaseDataFrameWriter
+from sqlframe.base.udf import _BaseUDFRegistration
 from sqlframe.base.util import (
     get_column_mapping_from_schema_input,
     normalize_string,
@@ -64,16 +65,18 @@ CATALOG = t.TypeVar("CATALOG", bound=_BaseCatalog)
 READER = t.TypeVar("READER", bound=_BaseDataFrameReader)
 WRITER = t.TypeVar("WRITER", bound=_BaseDataFrameWriter)
 DF = t.TypeVar("DF", bound=_BaseDataFrame)
+UDF_REGISTRATION = t.TypeVar("UDF_REGISTRATION", bound=_BaseUDFRegistration)
 
 _MISSING = "MISSING"
 
 
-class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
+class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN, UDF_REGISTRATION]):
     _instance = None
     _reader: t.Type[READER]
     _writer: t.Type[WRITER]
     _catalog: t.Type[CATALOG]
     _df: t.Type[DF]
+    _udf_registration: t.Type[UDF_REGISTRATION]
 
     SANITIZE_COLUMN_NAMES = False
 
@@ -81,7 +84,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
         self,
         conn: t.Optional[CONN] = None,
         schema: t.Optional[MappingSchema] = None,
-        case_sensitive: bool = False,
         *args,
         **kwargs,
     ):
@@ -91,11 +93,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
             self.execution_dialect: Dialect = Dialect.get_or_raise(
                 self.builder.DEFAULT_EXECUTION_DIALECT
             )
-            self.case_sensitive: bool = case_sensitive
-            if self.case_sensitive:
-                self.input_dialect.NORMALIZATION_STRATEGY = NormalizationStrategy.CASE_SENSITIVE
-                self.output_dialect.NORMALIZATION_STRATEGY = NormalizationStrategy.CASE_SENSITIVE
-                self.execution_dialect.NORMALIZATION_STRATEGY = NormalizationStrategy.CASE_SENSITIVE
             self.known_ids: t.Set[str] = set()
             self.known_branch_ids: t.Set[str] = set()
             self.known_sequence_ids: t.Set[str] = set()
@@ -176,6 +173,13 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
     def _has_connection(self) -> bool:
         return hasattr(self, "_connection") and bool(self._connection)
 
+    @property
+    def udf(self) -> UDF_REGISTRATION:
+        return self._udf_registration(self)
+
+    def getActiveSession(self) -> Self:
+        return self
+
     def range(self, *args):
         start = 0
         step = 1
@@ -573,53 +577,10 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
             converted_values.append(cls._to_value(value))
         return _create_row(columns, converted_values)
 
-    @property
-    def _is_standalone(self) -> bool:
-        from sqlframe.standalone.session import StandaloneSession
-
-        return isinstance(self, StandaloneSession)
-
-    @property
-    def _is_duckdb(self) -> bool:
-        from sqlframe.duckdb.session import DuckDBSession
-
-        return isinstance(self, DuckDBSession)
-
-    @property
-    def _is_postgres(self) -> bool:
-        from sqlframe.postgres.session import PostgresSession
-
-        return isinstance(self, PostgresSession)
-
-    @property
-    def _is_spark(self) -> bool:
-        from sqlframe.spark.session import SparkSession
-
-        return isinstance(self, SparkSession)
-
-    @property
-    def _is_bigquery(self) -> bool:
-        from sqlframe.bigquery.session import BigQuerySession
-
-        return isinstance(self, BigQuerySession)
-
-    @property
-    def _is_redshift(self) -> bool:
-        from sqlframe.redshift.session import RedshiftSession
-
-        return isinstance(self, RedshiftSession)
-
-    @property
-    def _is_snowflake(self) -> bool:
-        from sqlframe.snowflake.session import SnowflakeSession
-
-        return isinstance(self, SnowflakeSession)
-
     class Builder:
         SQLFRAME_INPUT_DIALECT_KEY = "sqlframe.input.dialect"
         SQLFRAME_OUTPUT_DIALECT_KEY = "sqlframe.output.dialect"
         SQLFRAME_EXECUTION_DIALECT_KEY = "sqlframe.execution.dialect"
-        SQLFRAME_CASE_SENSITIVE_KEY = "spark.sql.caseSensitive"
         SQLFRAME_CONN_KEY = "sqlframe.conn"
         SQLFRAME_SCHEMA_KEY = "sqlframe.schema"
         DEFAULT_INPUT_DIALECT = "spark"
@@ -665,8 +626,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
                     self._session_kwargs["conn"] = value
                 elif key == self.SQLFRAME_SCHEMA_KEY:
                     self._session_kwargs["schema"] = value
-                elif key == self.SQLFRAME_CASE_SENSITIVE_KEY:
-                    self._session_kwargs["case_sensitive"] = value
                 else:
                     self._session_kwargs[key] = value
             if map:
@@ -676,8 +635,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
                     self.output_dialect = map[self.SQLFRAME_OUTPUT_DIALECT_KEY]
                 if self.SQLFRAME_EXECUTION_DIALECT_KEY in map:
                     self.execution_dialect = map[self.SQLFRAME_EXECUTION_DIALECT_KEY]
-                if self.SQLFRAME_CASE_SENSITIVE_KEY in map:
-                    self._session_kwargs["case_sensitive"] = map[self.SQLFRAME_CASE_SENSITIVE_KEY]
                 if self.SQLFRAME_CONN_KEY in map:
                     self._session_kwargs["conn"] = map[self.SQLFRAME_CONN_KEY]
                 if self.SQLFRAME_SCHEMA_KEY in map:
@@ -700,15 +657,5 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
             self.session.execution_dialect = Dialect.get_or_raise(self.execution_dialect)
             if hasattr(self.session, "_connection") and not self.session._connection:
                 self.session._connection = self._conn
-            if self.session.case_sensitive:
-                self.session.input_dialect.NORMALIZATION_STRATEGY = (
-                    NormalizationStrategy.CASE_SENSITIVE
-                )
-                self.session.output_dialect.NORMALIZATION_STRATEGY = (
-                    NormalizationStrategy.CASE_SENSITIVE
-                )
-                self.session.execution_dialect.NORMALIZATION_STRATEGY = (
-                    NormalizationStrategy.CASE_SENSITIVE
-                )
 
     builder = Builder()

sqlframe/base/udf.py

@@ -0,0 +1,36 @@
+# This code is based on code from Apache Spark under the license found in the LICENSE file located in the 'sqlframe' folder.
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from sqlframe.base._typing import DataTypeOrString, UserDefinedFunctionLike
+    from sqlframe.base.session import _BaseSession
+
+    SESSION = t.TypeVar("SESSION", bound=_BaseSession)
+else:
+    SESSION = t.TypeVar("SESSION")
+
+
+class _BaseUDFRegistration(t.Generic[SESSION]):
+    def __init__(self, sparkSession: SESSION):
+        self.sparkSession = sparkSession
+
+    def register(
+        self,
+        name: str,
+        f: t.Union[t.Callable[..., t.Any], UserDefinedFunctionLike],
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> UserDefinedFunctionLike:
+        raise NotImplementedError
+
+    def registerJavaFunction(
+        self,
+        name: str,
+        javaClassName: str,
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> None:
+        raise NotImplementedError
+
+    def registerJavaUDAF(self, name: str, javaClassName: str) -> None:
+        raise NotImplementedError

sqlframe/bigquery/session.py

@@ -9,6 +9,7 @@ from sqlframe.bigquery.readwriter import (
     BigQueryDataFrameReader,
     BigQueryDataFrameWriter,
 )
+from sqlframe.bigquery.udf import BigQueryUDFRegistration
 
 if t.TYPE_CHECKING:
     from google.cloud.bigquery.client import Client as BigQueryClient
@@ -25,12 +26,14 @@ class BigQuerySession(
         BigQueryDataFrameWriter,
         BigQueryDataFrame,
         BigQueryConnection,
+        BigQueryUDFRegistration,
     ],
 ):
     _catalog = BigQueryCatalog
     _reader = BigQueryDataFrameReader
     _writer = BigQueryDataFrameWriter
     _df = BigQueryDataFrame
+    _udf_registration = BigQueryUDFRegistration
 
     QUALIFY_INFO_SCHEMA_WITH_DATABASE = True
     SANITIZE_COLUMN_NAMES = True

sqlframe/bigquery/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.bigquery.session import BigQuerySession
+
+
+class BigQueryUDFRegistration(_BaseUDFRegistration["BigQuerySession"]): ...

sqlframe/duckdb/session.py

@@ -11,6 +11,7 @@ from sqlframe.duckdb.readwriter import (
     DuckDBDataFrameReader,
     DuckDBDataFrameWriter,
 )
+from sqlframe.duckdb.udf import DuckDBUDFRegistration
 
 if t.TYPE_CHECKING:
     from duckdb import DuckDBPyConnection
@@ -26,12 +27,14 @@ class DuckDBSession(
         DuckDBDataFrameWriter,
         DuckDBDataFrame,
         DuckDBPyConnection,
+        DuckDBUDFRegistration,
     ]
 ):
     _catalog = DuckDBCatalog
     _reader = DuckDBDataFrameReader
     _writer = DuckDBDataFrameWriter
     _df = DuckDBDataFrame
+    _udf_registration = DuckDBUDFRegistration
 
     def __init__(self, conn: t.Optional[DuckDBPyConnection] = None, *args, **kwargs):
         import duckdb

sqlframe/duckdb/udf.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.base._typing import DataTypeOrString, UserDefinedFunctionLike
+    from sqlframe.duckdb.session import DuckDBSession
+
+
+class DuckDBUDFRegistration(_BaseUDFRegistration["DuckDBSession"]):
+    def register(  # type: ignore
+        self,
+        name: str,
+        f: t.Union[t.Callable[..., t.Any], UserDefinedFunctionLike],
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> UserDefinedFunctionLike:
+        self.sparkSession._conn.create_function(name, f, return_type=returnType)  # type: ignore

sqlframe/postgres/session.py

@@ -11,6 +11,7 @@ from sqlframe.postgres.readwriter import (
     PostgresDataFrameReader,
     PostgresDataFrameWriter,
 )
+from sqlframe.postgres.udf import PostgresUDFRegistration
 
 if t.TYPE_CHECKING:
     from psycopg2.extensions import connection as psycopg2_connection
@@ -27,12 +28,14 @@ class PostgresSession(
         PostgresDataFrameWriter,
         PostgresDataFrame,
         psycopg2_connection,
+        PostgresUDFRegistration,
     ],
 ):
     _catalog = PostgresCatalog
     _reader = PostgresDataFrameReader
     _writer = PostgresDataFrameWriter
     _df = PostgresDataFrame
+    _udf_registration = PostgresUDFRegistration
 
     def __init__(self, conn: t.Optional[psycopg2_connection] = None):
         if not hasattr(self, "_conn"):

sqlframe/postgres/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.postgres.session import PostgresSession
+
+
+class PostgresUDFRegistration(_BaseUDFRegistration["PostgresSession"]): ...

sqlframe/redshift/session.py

@@ -10,6 +10,7 @@ from sqlframe.redshift.readwriter import (
     RedshiftDataFrameReader,
     RedshiftDataFrameWriter,
 )
+from sqlframe.redshift.udf import RedshiftUDFRegistration
 
 if t.TYPE_CHECKING:
     from redshift_connector.core import Connection as RedshiftConnection
@@ -24,12 +25,14 @@ class RedshiftSession(
         RedshiftDataFrameWriter,
         RedshiftDataFrame,
         RedshiftConnection,
+        RedshiftUDFRegistration,
     ],
 ):
     _catalog = RedshiftCatalog
     _reader = RedshiftDataFrameReader
     _writer = RedshiftDataFrameWriter
     _df = RedshiftDataFrame
+    _udf_registration = RedshiftUDFRegistration
 
     def __init__(self, conn: t.Optional[RedshiftConnection] = None):
         warnings.warn(

sqlframe/redshift/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.redshift.session import RedshiftSession
+
+
+class RedshiftUDFRegistration(_BaseUDFRegistration["RedshiftSession"]): ...

sqlframe/snowflake/session.py

@@ -4,6 +4,8 @@ import json
 import typing as t
 import warnings
 
+from sqlframe.snowflake.udf import SnowflakeUDFRegistration
+
 try:
     from snowflake.connector.converter import SnowflakeConverter
 except ImportError:
@@ -50,12 +52,14 @@ class SnowflakeSession(
         SnowflakeDataFrameWriter,
         SnowflakeDataFrame,
         SnowflakeConnection,
+        SnowflakeUDFRegistration,
     ],
 ):
     _catalog = SnowflakeCatalog
     _reader = SnowflakeDataFrameReader
     _writer = SnowflakeDataFrameWriter
     _df = SnowflakeDataFrame
+    _udf_registration = SnowflakeUDFRegistration
 
     def __init__(self, conn: t.Optional[SnowflakeConnection] = None):
         import snowflake

sqlframe/snowflake/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.snowflake.session import SnowflakeSession
+
+
+class SnowflakeUDFRegistration(_BaseUDFRegistration["SnowflakeSession"]): ...

sqlframe/spark/session.py

@@ -14,6 +14,7 @@ from sqlframe.spark.readwriter import (
     SparkDataFrameWriter,
 )
 from sqlframe.spark.types import Row
+from sqlframe.spark.udf import SparkUDFRegistration
 
 if t.TYPE_CHECKING:
     import pandas as pd
@@ -32,12 +33,14 @@ class SparkSession(
         SparkDataFrameWriter,
         SparkDataFrame,
         PySparkSession,
+        SparkUDFRegistration,
     ],
 ):
     _catalog = SparkCatalog
     _reader = SparkDataFrameReader
     _writer = SparkDataFrameWriter
     _df = SparkDataFrame
+    _udf_registration = SparkUDFRegistration
 
     def __init__(self, conn: t.Optional[PySparkSession] = None, *args, **kwargs):
         from pyspark.sql.session import DataFrame, SparkSession

sqlframe/spark/udf.py

@@ -0,0 +1,34 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.base._typing import DataTypeOrString, UserDefinedFunctionLike
+    from sqlframe.spark.session import SparkSession  # type: ignore
+
+
+class SparkUDFRegistration(_BaseUDFRegistration["SparkSession"]):
+    def register(
+        self,
+        name: str,
+        f: t.Union[t.Callable[..., t.Any], UserDefinedFunctionLike],
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> UserDefinedFunctionLike:
+        return self.sparkSession.spark_session.udf.register(name, f, returnType=returnType)  # type: ignore
+
+    def registerJavaFunction(
+        self,
+        name: str,
+        javaClassName: str,
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> None:
+        self.sparkSession.spark_session.udf.registerJavaFunction(
+            name,
+            javaClassName,
+            returnType=returnType,  # type: ignore
+        )
+
+    def registerJavaUDAF(self, name: str, javaClassName: str) -> None:
+        self.sparkSession.spark_session.udf.registerJavaUDAF(name, javaClassName)

sqlframe/standalone/session.py

@@ -9,6 +9,7 @@ from sqlframe.standalone.readwriter import (
     StandaloneDataFrameReader,
     StandaloneDataFrameWriter,
 )
+from sqlframe.standalone.udf import StandaloneUDFRegistration
 
 
 class StandaloneSession(
@@ -18,12 +19,14 @@ class StandaloneSession(
         StandaloneDataFrameWriter,
         StandaloneDataFrame,
         object,
+        StandaloneUDFRegistration,
     ]
 ):  # type: ignore
     _catalog = StandaloneCatalog
     _reader = StandaloneDataFrameReader
     _writer = StandaloneDataFrameWriter
     _df = StandaloneDataFrame
+    _udf_registration = StandaloneUDFRegistration
 
     class Builder(_BaseSession.Builder):
         DEFAULT_INPUT_DIALECT = "spark"

sqlframe/standalone/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.standalone.session import StandaloneSession
+
+
+class StandaloneUDFRegistration(_BaseUDFRegistration["StandaloneSession"]): ...

{sqlframe-2.0.0.dist-info → sqlframe-2.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 2.0.0
+Version: 2.1.0
 Summary: Turning PySpark Into a Universal DataFrame API
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman
@@ -18,7 +18,7 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: prettytable (<3.11.0)
-Requires-Dist: sqlglot (<25.9,>=24.0.0)
+Requires-Dist: sqlglot (<25.11,>=24.0.0)
 Requires-Dist: typing-extensions (<5,>=4.8)
 Provides-Extra: bigquery
 Requires-Dist: google-cloud-bigquery-storage (<3,>=2) ; extra == 'bigquery'
@@ -26,7 +26,7 @@ Requires-Dist: google-cloud-bigquery[pandas] (<4,>=3) ; extra == 'bigquery'
 Provides-Extra: dev
 Requires-Dist: duckdb (<1.1,>=0.9) ; extra == 'dev'
 Requires-Dist: mypy (<1.12,>=1.10.0) ; extra == 'dev'
-Requires-Dist: openai (<1.38,>=1.30) ; extra == 'dev'
+Requires-Dist: openai (<1.41,>=1.30) ; extra == 'dev'
 Requires-Dist: pandas-stubs (<3,>=2) ; extra == 'dev'
 Requires-Dist: pandas (<3,>=2) ; extra == 'dev'
 Requires-Dist: psycopg (<4,>=3.1) ; extra == 'dev'
@@ -49,7 +49,7 @@ Provides-Extra: duckdb
 Requires-Dist: duckdb (<1.1,>=0.9) ; extra == 'duckdb'
 Requires-Dist: pandas (<3,>=2) ; extra == 'duckdb'
 Provides-Extra: openai
-Requires-Dist: openai (<1.38,>=1.30) ; extra == 'openai'
+Requires-Dist: openai (<1.41,>=1.30) ; extra == 'openai'
 Provides-Extra: pandas
 Requires-Dist: pandas (<3,>=2) ; extra == 'pandas'
 Provides-Extra: postgres

{sqlframe-2.0.0.dist-info → sqlframe-2.1.0.dist-info}/RECORD

@@ -1,21 +1,22 @@
 sqlframe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sqlframe/_version.py,sha256=o9XMlNC6GoiUn7mwoOOgJdijgp3B_2fLdgizcKs1i2o,411
+sqlframe/_version.py,sha256=N_eoCB5RSZeFANRJsHZ9FylE5ILLKrp6YmjX8ezqFpA,411
 sqlframe/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sqlframe/base/_typing.py,sha256=DuTay8-o9W-pw3RPZCgLunKNJLS9PkaV11G_pxXp9NY,1256
+sqlframe/base/_typing.py,sha256=b2clI5HI1zEZKB_3Msx3FeAJQyft44ubUifJwQRVXyQ,1298
 sqlframe/base/catalog.py,sha256=SzFQalTWdhWzxUY-4ut1f9TfOECp_JmJEgNPfrRKCe0,38457
 sqlframe/base/column.py,sha256=csci74Cdi7S7xpyAUfX6QVvlPDe4uR-S8Ejb50uCIfI,17698
 sqlframe/base/dataframe.py,sha256=OPDdOIlhVG8tytp94IR0qKPzEz3CDreTluB2NTV7Viw,70765
 sqlframe/base/decorators.py,sha256=Jy4bf8MhZ-AJ6CWTj59bBJRqamtLbPC0USUMFrY6g0w,449
 sqlframe/base/exceptions.py,sha256=9Uwvqn2eAkDpqm4BrRgbL61qM-GMCbJEMAW8otxO46s,370
 sqlframe/base/function_alternatives.py,sha256=VpBE1-p4_WifKoqeLJbZQFZ5MWJhNAEYCiQ2eE3hAOk,50015
-sqlframe/base/functions.py,sha256=nKnRTWf6NeyWVrctCFhqp9Ct-Ggu9t8JG0Ikma-dJ3o,189827
+sqlframe/base/functions.py,sha256=2dDfPepAuQvLcxwaZbj9qJeEeiqOaYJDI3vPoZXoM1Q,189959
 sqlframe/base/group.py,sha256=TES9CleVmH3x-0X-tqmuUKfCKSWjH5vg1aU3R6dDmFc,4059
 sqlframe/base/normalize.py,sha256=nXAJ5CwxVf4DV0GsH-q1w0p8gmjSMlv96k_ez1eVul8,3880
 sqlframe/base/operations.py,sha256=-AhNuEzcV7ZExoP1oY3blaKip-joQyJeQVvfBTs_2g4,3456
 sqlframe/base/readerwriter.py,sha256=FkImnUR_qNASmXktk0JDsFeOzfh799hd09vI2uznNH8,25350
-sqlframe/base/session.py,sha256=IG9rNUSCDR8d6m3-zLNHLL5S2zG6uvdRn7FA17cKi_M,27065
+sqlframe/base/session.py,sha256=YGlrc5nB2TOtrAzzgHynrkdRyl39DqczurnxyoqcfTg,25024
 sqlframe/base/transforms.py,sha256=y0j3SGDz3XCmNGrvassk1S-owllUWfkHyMgZlY6SFO4,467
 sqlframe/base/types.py,sha256=iBNk9bpFtb2NBIogYS8i7OlQZMRvpR6XxqzBebsjQDU,12280
+sqlframe/base/udf.py,sha256=O6hMhBUy9NVv-mhJRtfFhXTIa_-Z8Y_FkmmuOHu0l90,1117
 sqlframe/base/util.py,sha256=ktQfPBcYyXXLPWCOg9Gg_acjMc83xNykC8CFjuqGU5E,14879
 sqlframe/base/window.py,sha256=8hOv-ignPPIsZA9FzvYzcLE9J_glalVaYjIAUdRUX3o,4943
 sqlframe/base/mixins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -30,8 +31,9 @@ sqlframe/bigquery/functions.py,sha256=CxXiBzSLRf9h-9lAM8xSz1CFcVk2tVXU1eFPJ4d_L0
 sqlframe/bigquery/functions.pyi,sha256=BCYqHpZzv4KWVtTuiC9wCSzXdxeHsz9gwkEvKzwHnoY,13583
 sqlframe/bigquery/group.py,sha256=UVBNBRTo8OqS-_cS5YwvTeJYgYxeG-d6R3kfyHmlFqw,391
 sqlframe/bigquery/readwriter.py,sha256=WAD3ZMwkkjOpvPPoZXfaLLNM6tRTeUvdEj-hQZAzXeo,870
-sqlframe/bigquery/session.py,sha256=RU3Ompf71sLkW9miuYbMdA4HYTSra0-LI-rMCOa-9oQ,2623
+sqlframe/bigquery/session.py,sha256=-ySvZGbV-EeUJ6Z1SEFT8Es_80Lu3Db0WF333ktxoIQ,2762
 sqlframe/bigquery/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
+sqlframe/bigquery/udf.py,sha256=ZZ1-P1zWZhQqmhBqwAxfNeKl31nDkkZgkuz7Dn28P_0,264
 sqlframe/bigquery/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/duckdb/__init__.py,sha256=t85TA3ufZtL1weQNFmEs8itCSwbJFtw03-p0GT4XGf8,669
 sqlframe/duckdb/catalog.py,sha256=YYYVmetLUaJOdObKw4AJ7L0P-msshkta4xHlcZQ9zEA,4795
@@ -41,8 +43,9 @@ sqlframe/duckdb/functions.py,sha256=ZCmiHegr7484nNmTycxI9f5pDIDhIV3m5zTFK4hMD4Q,
 sqlframe/duckdb/functions.pyi,sha256=nU-6a2cfLDkuMCdYrNRLfa6-i8Aa0CxQQ1nLT6roIdI,5813
 sqlframe/duckdb/group.py,sha256=IkhbW42Ng1U5YT3FkIdiB4zBqRkW4QyTb-1detY1e_4,383
 sqlframe/duckdb/readwriter.py,sha256=0qZcARQoWYlx9P2m0uS2vuMj_tG_ka4NhHzg7qdaR3I,4597
-sqlframe/duckdb/session.py,sha256=WsPImstaJUrRec1RDb81fTE0t10wnNLS5685F1Ezj0k,2209
+sqlframe/duckdb/session.py,sha256=dj8kOTklBHKTyFGc3UwbgFlloPnfIDUf1Sh4uaO1hSg,2340
 sqlframe/duckdb/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
+sqlframe/duckdb/udf.py,sha256=Du9LnOtT1lJvB90D4HSR2tB7MXy179jZngDR-EjVjQk,656
 sqlframe/duckdb/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/postgres/__init__.py,sha256=Sz_MtgV_oh_QhfZTC7iKM07ICUmNcJEDV0kEkSW9ZKU,712
 sqlframe/postgres/catalog.py,sha256=9XVXXDW04mY_KmeB52NsCny8n0evqdDCBxhGH1Xce6s,8956
@@ -52,8 +55,9 @@ sqlframe/postgres/functions.py,sha256=ZWfb2sX3ZplT-tBsrsaXDc9lPzXoE-8ZflzP_FUfwD
 sqlframe/postgres/functions.pyi,sha256=um-qE2g9iPs0-53vJ46lArbfvDqAbFIwrxLJgcrPM_8,5536
 sqlframe/postgres/group.py,sha256=KUXeSFKWTSH9yCRJAhW85OvjZaG6Zr4In9LR_ie3yGU,391
 sqlframe/postgres/readwriter.py,sha256=L1e3yKXzFVNR_W5s1DHaWol7G8x7l4jcZ5sLGualyMk,870
-sqlframe/postgres/session.py,sha256=C7cYx3W6QuSCyfZ4x2ELyED566Sxc0kjEjz6uv6NA0w,2356
+sqlframe/postgres/session.py,sha256=LkM35-ADkIh2OSNT6HRVfJ8jLOoFlrdgkSSVhFGMYRc,2495
 sqlframe/postgres/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
+sqlframe/postgres/udf.py,sha256=TylVxrmPzycAqnpTiueGvvetiMCfCX31QatgQMIgpME,264
 sqlframe/postgres/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/redshift/__init__.py,sha256=jamKYQtQaKjjXnQ01QGPHvatbrZSw9sWno_VOUGSz6I,712
 sqlframe/redshift/catalog.py,sha256=x-sfVwoOS_0MeLICCqu7YUiFRgBam7vV1sHfaAKZGjE,5410
@@ -62,8 +66,9 @@ sqlframe/redshift/dataframe.py,sha256=aTC0DOPDFwWH1_b9T0Pif80cYSGudIp0D-cmkR7Ci_
 sqlframe/redshift/functions.py,sha256=DR5kodYAcKatUqopwrEQtxryI4ZSqaH47_y3WLht4Wg,455
 sqlframe/redshift/group.py,sha256=5MGZYJfHpzoRSQ0N_pn4KUk4Mk2gocQwU3K1-jAbvGg,391
 sqlframe/redshift/readwriter.py,sha256=g3FYKSsJKqcSnElprzzz29ZctoXq9tRB0Mj9Bm1HycI,870
-sqlframe/redshift/session.py,sha256=Hy49PMCHjY6c_BriW43yIp0mfjNmUp8MbCFx6cFYn3w,1454
+sqlframe/redshift/session.py,sha256=nqqFxplCcIa_mN59xq3yHclMQ_dKchwuodx31xog55o,1593
 sqlframe/redshift/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
+sqlframe/redshift/udf.py,sha256=xOi7vtO56MyYprC3ys3FCSVJ5EVjEq2l_KqEVgKJrkw,264
 sqlframe/redshift/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/snowflake/__init__.py,sha256=nuQ3cuHjDpW4ELZfbd2qOYmtXmcYl7MtsrdOrRdozo0,746
 sqlframe/snowflake/catalog.py,sha256=Tp_B9oMTInnqLTiZNsT0UNc-a-aKXu3zFThlCwLeKCA,6507
@@ -73,8 +78,9 @@ sqlframe/snowflake/functions.py,sha256=iX67xU8Czy5k3HiNjNbN-x-via6yqGNaMPw1maV9H
 sqlframe/snowflake/functions.pyi,sha256=wqUPXuZxjRY0rPY2BRAb7XXkWYP1DyuDzvlriRySdSw,6185
 sqlframe/snowflake/group.py,sha256=pPP1l2RRo_LgkXrji8a87n2PKo-63ZRPT-WUtvVcBME,395
 sqlframe/snowflake/readwriter.py,sha256=yhRc2HcMq6PwV3ghZWC-q-qaE7LE4aEjZEXCip4OOlQ,884
-sqlframe/snowflake/session.py,sha256=YtcoKz4vII6NmC3hWahQlthImoCbzQO37Ww0-ZzSGDk,3194
+sqlframe/snowflake/session.py,sha256=pCTb39olI5fzj7mrnuWupJs5EenDVDSH9goxueTXwiU,3338
 sqlframe/snowflake/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
+sqlframe/snowflake/udf.py,sha256=yzMmky-n5BXFbdldgfzLP8hCrVm0DgruSUCUUy1_3sk,268
 sqlframe/snowflake/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/spark/__init__.py,sha256=WhYQAZMJN1EMNAVGUH7BEinxNdYtXOrrr-6HUniJOyI,649
 sqlframe/spark/catalog.py,sha256=3T4aivgQmAhPTmlVk53jk3VvFBJlp1pKvWKNmTHz1W8,39741
@@ -84,8 +90,9 @@ sqlframe/spark/functions.py,sha256=AQfqfvaojJzCuo9DyDklz0JYZPhn_3dzWvztsklBO0o,5
 sqlframe/spark/functions.pyi,sha256=bjz6s8E6OB0c4KfTTsls7rhb_R9mIYvkaeaXefMziqM,11617
 sqlframe/spark/group.py,sha256=MrvV_v-YkBc6T1zz882WrEqtWjlooWIyHBCmTQg3fCA,379
 sqlframe/spark/readwriter.py,sha256=w68EImTcGJv64X7pc1tk5tDjDxb1nAnn-MiIaaN9Dc8,812
-sqlframe/spark/session.py,sha256=ADein7FMCutOriYVhQVQFMW--rdbuk-Y7Wzervs4DW0,5156
+sqlframe/spark/session.py,sha256=eYwEANEMkP6djzUlm-p1WPkB6QZ4cdW_P4sx1K1wOwQ,5283
 sqlframe/spark/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
+sqlframe/spark/udf.py,sha256=owB8NDaGVkUQ0WGm7SZt2t9zfvLFCfi0W48QiPfgjck,1153
 sqlframe/spark/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/standalone/__init__.py,sha256=yu4A97HwhyDwllDEzG7io4ScyWipWSAH2tqUKS545OA,767
 sqlframe/standalone/catalog.py,sha256=oJAPxrXtra_YP_JBZCJY2qsr0TRhWG7FFSq0RHDszcU,389
@@ -94,13 +101,14 @@ sqlframe/standalone/dataframe.py,sha256=o6weWJvDX1w351p_14PQ52qopbPozVG9OFmhaqVv
 sqlframe/standalone/functions.py,sha256=NW-k7NP_Y9DzQq6fjX5-CL8oOUGAiaMo4SBuDd8-JUA,38
 sqlframe/standalone/group.py,sha256=oGEbAQMSm6AlkwnBxNI8r9enZWRwsRxc8zpzoz3rArk,399
 sqlframe/standalone/readwriter.py,sha256=EZNyDJ4ID6sGNog3uP4-e9RvchX4biJJDNtc5hkKkrY,633
-sqlframe/standalone/session.py,sha256=wQmdu2sv6KMTAv0LRFk7TY7yzlh3xvmsyqilEtRecbY,1191
+sqlframe/standalone/session.py,sha256=h8EpjZOt4GiDBkqEC5dFdkv5PtCDF9TkD9spqp4olSE,1338
 sqlframe/standalone/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
+sqlframe/standalone/udf.py,sha256=azmgtUjHNIPs0WMVNId05SHwiYn41MKVBhKXsQJ5dmY,272
 sqlframe/standalone/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/testing/__init__.py,sha256=VVCosQhitU74A3NnE52O4mNtGZONapuEXcc20QmSlnQ,132
 sqlframe/testing/utils.py,sha256=9DDYVuocO7tygee3RaajuJNZ24sJwf_LY556kKg7kTw,13011
-sqlframe-2.0.0.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
-sqlframe-2.0.0.dist-info/METADATA,sha256=Eluga2lgMiRawpetmiV1JF5aV9kxRvqK-F_4NKaKN7Q,7809
-sqlframe-2.0.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-sqlframe-2.0.0.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
-sqlframe-2.0.0.dist-info/RECORD,,
+sqlframe-2.1.0.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
+sqlframe-2.1.0.dist-info/METADATA,sha256=LkxVdNzdiCYsXSIsY-SSZ2qDq9RWvqA2_-cBS5p_N8Q,7810
+sqlframe-2.1.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+sqlframe-2.1.0.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
+sqlframe-2.1.0.dist-info/RECORD,,