gcdZddlmZddlmZddlmZddlmZddlmZerddl m Z ee e fZ neZ dgZ Gd de Zy ) zxSchema. Adapted from Polars implementation at: https://github.com/pola-rs/polars/blob/main/py-polars/polars/schema.py. ) annotations) OrderedDict) TYPE_CHECKING)Iterable)Mapping)DTypeSchemacFeZdZdZ d dfd ZddZd dZd dZxZS) r aOrdered mapping of column names to their data type. Arguments: schema: Mapping[str, DType] | Iterable[tuple[str, DType]] | None The schema definition given by column names and their associated. *instantiated* Narwhals data type. Accepts a mapping or an iterable of tuples. Examples: Define a schema by passing *instantiated* data types. >>> import narwhals as nw >>> schema = nw.Schema({"foo": nw.Int8(), "bar": nw.String()}) >>> schema Schema({'foo': Int8, 'bar': String}) Access the data type associated with a specific column name. >>> schema["foo"] Int8 Access various schema properties using the `names`, `dtypes`, and `len` methods. >>> schema.names() ['foo', 'bar'] >>> schema.dtypes() [Int8, String] >>> schema.len() 2 c0|xsi}t||yN)super__init__)selfschema __class__s D/var/www/openai/venv/lib/python3.12/site-packages/narwhals/schema.pyrzSchema.__init__8s2  c4t|jS)zXGet the column names of the schema. Returns: Column names. )listkeysrs rnamesz Schema.names>s DIIK  rc4t|jS)z^Get the data types of the schema. Returns: Data types of schema. )rvaluesrs rdtypesz Schema.dtypesFs DKKM""rct|S)zbGet the number of columns in the schema. Returns: Number of columns. )lenrs rrz Schema.lenNs 4yrr )rz8Mapping[str, DType] | Iterable[tuple[str, DType]] | NonereturnNone)rz list[str])rz list[DType])rint) __name__ __module__ __qualname____doc__rrrr __classcell__)rs@rr r s3>RV!N! ! !#rN)r$ __future__r collectionsrtypingrrrnarwhals.dtypesrstr BaseSchema__all__r rrr.sI ## %S%Z(JJ *;Z;r