additory 0.1.1a1__py3-none-any.whl → 0.1.1a2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. additory/__init__.py +1 -1
  2. additory/common/result.py +11 -1
  3. {additory-0.1.1a1.dist-info → additory-0.1.1a2.dist-info}/METADATA +2 -1
  4. {additory-0.1.1a1.dist-info → additory-0.1.1a2.dist-info}/RECORD +6 -6
  5. {additory-0.1.1a1.dist-info → additory-0.1.1a2.dist-info}/WHEEL +0 -0
  6. {additory-0.1.1a1.dist-info → additory-0.1.1a2.dist-info}/top_level.txt +0 -0
additory/__init__.py CHANGED
@@ -57,7 +57,7 @@ add = SimpleNamespace(
57
57
  )
58
58
 
59
59
  # Version
60
- __version__ = "0.1.1a1"
60
+ __version__ = "0.1.1a2"
61
61
 
62
62
  # Public API
63
63
  __all__ = ['add', '__version__']
additory/common/result.py CHANGED
@@ -43,7 +43,17 @@ class DataFrameResult:
43
43
 
44
44
  # Extract common metadata
45
45
  self.input_shape = metadata.get('input_shape', (0, 0))
46
- self.output_shape = (df.height, df.width)
46
+
47
+ # Get shape in a backend-agnostic way
48
+ if hasattr(df, 'height') and hasattr(df, 'width'):
49
+ # Polars DataFrame
50
+ self.output_shape = (df.height, df.width)
51
+ elif hasattr(df, 'shape'):
52
+ # pandas/cuDF DataFrame
53
+ self.output_shape = df.shape
54
+ else:
55
+ self.output_shape = (0, 0)
56
+
47
57
  self.columns_added = metadata.get('columns_added', [])
48
58
  self.columns_removed = metadata.get('columns_removed', [])
49
59
  self.execution_time = metadata.get('execution_time', 0.0)
{additory-0.1.1a1.dist-info → additory-0.1.1a2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: additory
3
- Version: 0.1.1a1
3
+ Version: 0.1.1a2
4
4
  Summary: Data augmentation library with Polars backend
5
5
  Author-email: Additory Team <team@additory.dev>
6
6
  License: MIT
@@ -22,6 +22,7 @@ Description-Content-Type: text/markdown
22
22
  Requires-Dist: polars>=0.20.0
23
23
  Requires-Dist: pandas>=2.0.0
24
24
  Requires-Dist: numpy>=1.24.0
25
+ Requires-Dist: pyarrow>=14.0.0
25
26
  Provides-Extra: dev
26
27
  Requires-Dist: pytest>=7.0.0; extra == "dev"
27
28
  Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
{additory-0.1.1a1.dist-info → additory-0.1.1a2.dist-info}/RECORD RENAMED
@@ -1,10 +1,10 @@
1
- additory/__init__.py,sha256=Ogj8nwpu_ERDNTNVGVxCLpI2O8LaPRI6YD9VC3mugVM,1810
1
+ additory/__init__.py,sha256=oRtcL1DpcVdfjcoRjPsRleGHEquxXYCy3siQDIYUn40,1810
2
2
  additory/common/__init__.py,sha256=b70Ksk5yOw5Lb1hyO3m0iEzoMGnBnxKAHkof9s1B7yw,957
3
3
  additory/common/column_selector.py,sha256=lWoK7sbrUy9m7F9Ya3zKjAvt7LCqJDjWfqv_qblAwqQ,6944
4
4
  additory/common/distributions.py,sha256=aE8oXUXKdtJooeLiRALV2l40lMkxRghvb4bIW30b4mM,11245
5
5
  additory/common/extractors.py,sha256=j10qoEXFWlKfCo0J7HfTC9dbE6eJefFUPaMAtIuQ_No,10756
6
6
  additory/common/knn_imputation.py,sha256=Z0g5MAc88_yqDEph-pP20k8Njw83CA7RmFErTF478Lo,10756
7
- additory/common/result.py,sha256=ug68SLnP9qS4qayYhWpFRvSfbz-hJV_GxrrFPLojdqE,12221
7
+ additory/common/result.py,sha256=PQf3htWgANuBR-xJH0WFxqy0ASy7tD5nalW2d-uV5P0,12548
8
8
  additory/common/strategy_parser.py,sha256=PE6kaW8JNOGTPOdxbGfCp8LBvK03USfInAt-C1LiOSw,8226
9
9
  additory/common/unit_conversions.py,sha256=H3gaJTFFeIQJK3B7Xo23TX4cA0HfuPAWa79tuT5MciQ,9558
10
10
  additory/common/validation.py,sha256=oOnPR_kQ8Eh5oWyd9z33UglYjgMy0HXV8HOBFqJdV94,9862
@@ -56,7 +56,7 @@ additory/functions/transform/harmonize.py,sha256=TY0ZYedhGh95GVpbG1-EgauT2e__amB
56
56
  additory/functions/transform/knn.py,sha256=wWay5L8Vgg8nDf_HwzJ7PSa27RdsvfvKLUqYw1125Gk,1676
57
57
  additory/functions/transform/onehotencoding.py,sha256=LibYOfphxrESKNNm-XTByBkjOhaKeitTfvO5mVVC8Rc,1891
58
58
  additory/functions/transform/transpose.py,sha256=Jy-50GgQbBsidBHY-F9YIBmwai1bPSSNzkN5X8n-sUk,1177
59
- additory-0.1.1a1.dist-info/METADATA,sha256=R2A_EU0OTYn5OFJUrSPQfrDgAzs8YWta0Z0TY6b5jn4,2432
60
- additory-0.1.1a1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
61
- additory-0.1.1a1.dist-info/top_level.txt,sha256=4zphwXiI6HEl40fdjMXoUp9JNIqQ-tgYWeo3zqKqvEk,9
62
- additory-0.1.1a1.dist-info/RECORD,,
59
+ additory-0.1.1a2.dist-info/METADATA,sha256=ND4TsR0rfm-a6pn2BOAILQtVN-vzOnx9-d6fNKsgoxY,2463
60
+ additory-0.1.1a2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
61
+ additory-0.1.1a2.dist-info/top_level.txt,sha256=4zphwXiI6HEl40fdjMXoUp9JNIqQ-tgYWeo3zqKqvEk,9
62
+ additory-0.1.1a2.dist-info/RECORD,,