pydiverse-common 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

pydiverse/common/__init__.py

@@ -20,10 +20,10 @@ from .dtypes import (
     PandasBackend,
     String,
     Time,
-    Uint8,
-    Uint16,
-    Uint32,
-    Uint64,
+    UInt8,
+    UInt16,
+    UInt32,
+    UInt64,
 )
 
 __all__ = [
@@ -44,10 +44,10 @@ __all__ = [
     "NullType",
     "String",
     "Time",
-    "Uint8",
-    "Uint16",
-    "Uint32",
-    "Uint64",
+    "UInt8",
+    "UInt16",
+    "UInt32",
+    "UInt64",
     "List",
     "PandasBackend",
 ]

pydiverse/common/dtypes.py

@@ -102,13 +102,13 @@ class Dtype:
             raise TypeError
         if pd.api.types.is_unsigned_integer_dtype(pandas_type):
             if is_np_dtype(pandas_type, np.uint64):
-                return Uint64()
+                return UInt64()
             elif is_np_dtype(pandas_type, np.uint32):
-                return Uint32()
+                return UInt32()
             elif is_np_dtype(pandas_type, np.uint16):
-                return Uint16()
+                return UInt16()
             elif is_np_dtype(pandas_type, np.uint8):
-                return Uint8()
+                return UInt8()
             raise TypeError
         if pd.api.types.is_float_dtype(pandas_type):
             if is_np_dtype(pandas_type, np.float64):
@@ -143,13 +143,13 @@ class Dtype:
             raise TypeError
         if pa.types.is_unsigned_integer(arrow_type):
             if pa.types.is_uint64(arrow_type):
-                return Uint64()
+                return UInt64()
             if pa.types.is_uint32(arrow_type):
-                return Uint32()
+                return UInt32()
             if pa.types.is_uint16(arrow_type):
-                return Uint16()
+                return UInt16()
             if pa.types.is_uint8(arrow_type):
-                return Uint8()
+                return UInt8()
             raise TypeError
         if pa.types.is_floating(arrow_type):
             if pa.types.is_float64(arrow_type):
@@ -186,10 +186,10 @@ class Dtype:
             pl.Int32: Int32(),
             pl.Int16: Int16(),
             pl.Int8: Int8(),
-            pl.UInt64: Uint64(),
-            pl.UInt32: Uint32(),
-            pl.UInt16: Uint16(),
-            pl.UInt8: Uint8(),
+            pl.UInt64: UInt64(),
+            pl.UInt32: UInt32(),
+            pl.UInt16: UInt16(),
+            pl.UInt8: UInt8(),
             pl.Float64: Float64(),
             pl.Float32: Float32(),
             pl.Decimal: Decimal(),
@@ -207,14 +207,16 @@ class Dtype:
         import sqlalchemy as sqa
 
         return {
+            Int(): sqa.BigInteger(),  # we default to 64 bit
             Int8(): sqa.SmallInteger(),
             Int16(): sqa.SmallInteger(),
             Int32(): sqa.Integer(),
             Int64(): sqa.BigInteger(),
-            Uint8(): sqa.SmallInteger(),
-            Uint16(): sqa.Integer(),
-            Uint32(): sqa.BigInteger(),
-            Uint64(): sqa.BigInteger(),
+            UInt8(): sqa.SmallInteger(),
+            UInt16(): sqa.Integer(),
+            UInt32(): sqa.BigInteger(),
+            UInt64(): sqa.BigInteger(),
+            Float(): sqa.Float(53),  # we default to 64 bit
             Float32(): sqa.Float(24),
             Float64(): sqa.Float(53),
             Decimal(): sqa.DECIMAL(),
@@ -245,14 +247,16 @@ class Dtype:
             raise TypeError("pandas doesn't have a native time dtype")
 
         return {
+            Int(): pd.Int64Dtype(),  # we default to 64 bit
             Int8(): pd.Int8Dtype(),
             Int16(): pd.Int16Dtype(),
             Int32(): pd.Int32Dtype(),
             Int64(): pd.Int64Dtype(),
-            Uint8(): pd.UInt8Dtype(),
-            Uint16(): pd.UInt16Dtype(),
-            Uint32(): pd.UInt32Dtype(),
-            Uint64(): pd.UInt64Dtype(),
+            UInt8(): pd.UInt8Dtype(),
+            UInt16(): pd.UInt16Dtype(),
+            UInt32(): pd.UInt32Dtype(),
+            UInt64(): pd.UInt64Dtype(),
+            Float(): pd.Float64Dtype(),  # we default to 64 bit
             Float32(): pd.Float32Dtype(),
             Float64(): pd.Float64Dtype(),
             Decimal(): pd.Float64Dtype(),  # NumericDtype is
@@ -267,14 +271,16 @@ class Dtype:
         import pyarrow as pa
 
         return {
+            Int(): pa.int64(),  # we default to 64 bit
             Int8(): pa.int8(),
             Int16(): pa.int16(),
             Int32(): pa.int32(),
             Int64(): pa.int64(),
-            Uint8(): pa.uint8(),
-            Uint16(): pa.uint16(),
-            Uint32(): pa.uint32(),
-            Uint64(): pa.uint64(),
+            UInt8(): pa.uint8(),
+            UInt16(): pa.uint16(),
+            UInt32(): pa.uint32(),
+            UInt64(): pa.uint64(),
+            Float(): pa.float64(),  # we default to 64 bit
             Float32(): pa.float32(),
             Float64(): pa.float64(),
             Decimal(): pa.decimal128(35, 10),  # Arbitrary precision
@@ -289,14 +295,16 @@ class Dtype:
         import polars as pl
 
         return {
+            Int(): pl.Int64,  # we default to 64 bit
             Int64(): pl.Int64,
             Int32(): pl.Int32,
             Int16(): pl.Int16,
             Int8(): pl.Int8,
-            Uint64(): pl.UInt64,
-            Uint32(): pl.UInt32,
-            Uint16(): pl.UInt16,
-            Uint8(): pl.UInt8,
+            UInt64(): pl.UInt64,
+            UInt32(): pl.UInt32,
+            UInt16(): pl.UInt16,
+            UInt8(): pl.UInt8,
+            Float(): pl.Float64,  # we default to 64 bit
             Float64(): pl.Float64,
             Float32(): pl.Float32,
             Decimal(): pl.Decimal(scale=10),  # Arbitrary precision
@@ -343,16 +351,16 @@ class Int16(Int): ...
 class Int8(Int): ...
 
 
-class Uint64(Int): ...
+class UInt64(Int): ...
 
 
-class Uint32(Int): ...
+class UInt32(Int): ...
 
 
-class Uint16(Int): ...
+class UInt16(Int): ...
 
 
-class Uint8(Int): ...
+class UInt8(Int): ...
 
 
 class String(Dtype): ...

{pydiverse_common-0.2.0.dist-info → pydiverse_common-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydiverse-common
-Version: 0.2.0
+Version: 0.3.0
 Summary: Common functionality shared between pydiverse libraries
 Author: QuantCo, Inc.
 Author-email: Martin Trautmann <windiana@users.sf.net>, Finn Rudolph <finn.rudolph@t-online.de>

{pydiverse_common-0.2.0.dist-info → pydiverse_common-0.3.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
-pydiverse/common/__init__.py,sha256=dMbOmWvQjkFv0agRnVGxvShqLzIR3ItAfshmbStoF7M,732
-pydiverse/common/dtypes.py,sha256=mh7PDQ3R7JkcHkNy83R8Z8bjdOxSlbyqEsBDGFfe0E0,11605
+pydiverse/common/__init__.py,sha256=2t3wnZGY_KITo5lysb-GAgXizeyd_iZvPjEHKnMsYpA,732
+pydiverse/common/dtypes.py,sha256=7qPxKH1yqVSacANi5Kl36DdBTHKSMZI2WkZCiEEejyU,12074
 pydiverse/common/errors/__init__.py,sha256=FNeEfVbUa23b9sHkFsmxHYhY6sRgjaZysPQmlovpJrI,262
 pydiverse/common/util/__init__.py,sha256=fGdKZtLaTVBW7NfCpX7rZhKHwUzmBnsuY2akDOnAnjc,315
 pydiverse/common/util/computation_tracing.py,sha256=HeXRHRUI8vxpzQ27Xcpa0StndSTP63EMT9vj4trPJUY,9697
@@ -9,7 +9,7 @@ pydiverse/common/util/disposable.py,sha256=4XoGz70YRWA9TAqnUBvRCTAdsOGBviFN0gzxU
 pydiverse/common/util/hashing.py,sha256=6x77BKg-w61u59fuTe9di0BtU-kEKH6UTRcKsRoYJ84,1196
 pydiverse/common/util/import_.py,sha256=K7dSgz4YyrqEvqhoOzbwgD7D8HScMoO5XoSWtjbaoUs,4056
 pydiverse/common/util/structlog.py,sha256=g0d8yaXBzAxmGNGZYMnMP9dsSQ__jN44GAY8Mb0ABeI,3487
-pydiverse_common-0.2.0.dist-info/METADATA,sha256=yxg3KMrzFkha_sNRIO3jCLeXxyq9xkf1alW6vT584jE,3357
-pydiverse_common-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pydiverse_common-0.2.0.dist-info/licenses/LICENSE,sha256=AcE6SDVuAq6v9ZLE_8eOCe_NvSE0rAPR3NR7lSowYh4,1517
-pydiverse_common-0.2.0.dist-info/RECORD,,
+pydiverse_common-0.3.0.dist-info/METADATA,sha256=5pwbOinMHcsH1Nbhe69cbh0eouevUvFNGt-QlLdl-HU,3357
+pydiverse_common-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pydiverse_common-0.3.0.dist-info/licenses/LICENSE,sha256=AcE6SDVuAq6v9ZLE_8eOCe_NvSE0rAPR3NR7lSowYh4,1517
+pydiverse_common-0.3.0.dist-info/RECORD,,