Method read_sql

sdk/python/feast/infra/ray_initializer.py:252–263

Read from a SQL database. Builds the connection factory from connection_url.

(self, sql: str, connection_url: str, **kwargs)

        )

    def read_sql(self, sql: str, connection_url: str, **kwargs) -> Any:
        """Read from a SQL database.  Builds the connection factory from connection_url."""
        import sqlalchemy

        def _connection_factory():
            # raw_connection() returns a DB API2-compliant connection (with .cursor())
            # from the underlying driver.  engine.connect() returns a SQLAlchemy 2.0
            # Connection object which does NOT expose .cursor(), failing Ray Data's
            # DB API2 compliance check.
            return sqlalchemy.create_engine(connection_url).raw_connection()

        return ray.data.read_sql(sql, _connection_factory, **kwargs)

    def from_huggingface(
        self, dataset_name: str, split: str = "train", **kwargs