aspect-data 0.0.2__tar.gz → 0.0.3__tar.gz

{aspect_data-0.0.2 → aspect_data-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspect-data
-Version: 0.0.2
+Version: 0.0.3
 Summary: General purpose, chemistry, and taxonomic featurization of tabular data with caching.
 Author-email: Eachan Johnson <eachan.johnson@crick.ac.uk>
 License: MIT License
@@ -234,7 +234,7 @@ pip install "aspect-data[bio]"
 Install transformer support if you want Hugging Face model embeddings:
 
 ```bash
-pip install "aspect[deep]"
+pip install "aspect-data[deep]"
 ```
 
 ### Development installation
@@ -507,7 +507,7 @@ pipe = DataPipeline({
 Require:
 
 ```bash
-pip install "aspect[chem]"
+pip install "aspect-data[chem]"
 ```
 
 | Transform            |  Input |                   Output |
@@ -542,7 +542,7 @@ features = pipe(data, drop_unused_columns=True)
 Require:
 
 ```bash
-pip install "aspect[bio]"
+pip install "aspect-data[bio]"
 ```
 
 | Transform             |                    Input |                Output |
@@ -577,7 +577,7 @@ features = pipe(data)
 Require:
 
 ```bash
-pip install "aspect[deep]"
+pip install "aspect-data[deep]"
 ```
 
 | Transform | Input |                                    Output |
@@ -620,7 +620,7 @@ The model will be loaded through Hugging Face `transformers`, so the first run m
 Require:
 
 ```bash
-pip install "aspect[chemprop]"
+pip install "aspect-data[chemprop]"
 ```
 
 `chemprop-mol` is intended for converting SMILES into Chemprop-style molecular datapoints, with optional labels and extra dense features.

{aspect_data-0.0.2 → aspect_data-0.0.3}/README.md

@@ -172,7 +172,7 @@ pip install "aspect-data[bio]"
 Install transformer support if you want Hugging Face model embeddings:
 
 ```bash
-pip install "aspect[deep]"
+pip install "aspect-data[deep]"
 ```
 
 ### Development installation
@@ -445,7 +445,7 @@ pipe = DataPipeline({
 Require:
 
 ```bash
-pip install "aspect[chem]"
+pip install "aspect-data[chem]"
 ```
 
 | Transform            |  Input |                   Output |
@@ -480,7 +480,7 @@ features = pipe(data, drop_unused_columns=True)
 Require:
 
 ```bash
-pip install "aspect[bio]"
+pip install "aspect-data[bio]"
 ```
 
 | Transform             |                    Input |                Output |
@@ -515,7 +515,7 @@ features = pipe(data)
 Require:
 
 ```bash
-pip install "aspect[deep]"
+pip install "aspect-data[deep]"
 ```
 
 | Transform | Input |                                    Output |
@@ -558,7 +558,7 @@ The model will be loaded through Hugging Face `transformers`, so the first run m
 Require:
 
 ```bash
-pip install "aspect[chemprop]"
+pip install "aspect-data[chemprop]"
 ```
 
 `chemprop-mol` is intended for converting SMILES into Chemprop-style molecular datapoints, with optional labels and extra dense features.

{aspect_data-0.0.2 → aspect_data-0.0.3}/aspect/data.py

@@ -401,11 +401,15 @@ class DataPipeline:
     def _unsqueeze(
         x: Mapping[str, ArrayLike]
     ) -> Dict[str, np.ndarray]:
-        for key in x:
-            vals = np.asarray(x[key])
-            if vals.ndim == 1:
-                x[key] = vals[:, None]
-        return x
+        def _unsqueeze(x, columns=None):
+            columns = columns or x.keys()
+            for key in columns:
+                vals = x[key]
+                if not isinstance(vals, dict):
+                    vals = np.asarray(x[key])
+                    if vals.ndim == 1 and np.issubdtype(vals.dtype, np.number):
+                        x[key] = vals[:, None]
+            return x
 
     def __call__(
         self, 

{aspect_data-0.0.2 → aspect_data-0.0.3}/aspect/transform/base.py

@@ -166,7 +166,7 @@ class ColumnTransform:
         inputs: Mapping[str, Iterable]
     ) -> np.ndarray:
         result = self.function(inputs, self.input_column)
-        if result.ndim == 1:
+        if hasattr(result, "ndim") and result.ndim == 1:
             result = result[:, None]
         return result
 

{aspect_data-0.0.2 → aspect_data-0.0.3}/aspect_data.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspect-data
-Version: 0.0.2
+Version: 0.0.3
 Summary: General purpose, chemistry, and taxonomic featurization of tabular data with caching.
 Author-email: Eachan Johnson <eachan.johnson@crick.ac.uk>
 License: MIT License
@@ -234,7 +234,7 @@ pip install "aspect-data[bio]"
 Install transformer support if you want Hugging Face model embeddings:
 
 ```bash
-pip install "aspect[deep]"
+pip install "aspect-data[deep]"
 ```
 
 ### Development installation
@@ -507,7 +507,7 @@ pipe = DataPipeline({
 Require:
 
 ```bash
-pip install "aspect[chem]"
+pip install "aspect-data[chem]"
 ```
 
 | Transform            |  Input |                   Output |
@@ -542,7 +542,7 @@ features = pipe(data, drop_unused_columns=True)
 Require:
 
 ```bash
-pip install "aspect[bio]"
+pip install "aspect-data[bio]"
 ```
 
 | Transform             |                    Input |                Output |
@@ -577,7 +577,7 @@ features = pipe(data)
 Require:
 
 ```bash
-pip install "aspect[deep]"
+pip install "aspect-data[deep]"
 ```
 
 | Transform | Input |                                    Output |
@@ -620,7 +620,7 @@ The model will be loaded through Hugging Face `transformers`, so the first run m
 Require:
 
 ```bash
-pip install "aspect[chemprop]"
+pip install "aspect-data[chemprop]"
 ```
 
 `chemprop-mol` is intended for converting SMILES into Chemprop-style molecular datapoints, with optional labels and extra dense features.

{aspect_data-0.0.2 → aspect_data-0.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aspect-data"
-version = "0.0.2"
+version = "0.0.3"
 authors = [
   { name="Eachan Johnson", email="eachan.johnson@crick.ac.uk" },
 ]