upgini 1.1.110a1__py3-none-any.whl → 1.1.111__py3-none-any.whl

upgini/features_enricher.py

@@ -131,6 +131,8 @@ class FeaturesEnricher(TransformerMixin):
         random_state: int = 42,
         cv: Optional[CVType] = None,
         detect_missing_search_keys: bool = True,
+        generate_features: Optional[List[str]] = None,
+        round_embeddings: Optional[int] = None,
         logs_enabled: bool = True,
         **kwargs,
     ):
@@ -203,8 +205,24 @@ class FeaturesEnricher(TransformerMixin):
             if self.runtime_parameters is None:
                 self.runtime_parameters = RuntimeParameters()
             if self.runtime_parameters.properties is None:
-                self.runtime_parameters.properties = {}
+                self.runtime_parameters.properties = dict()
             self.runtime_parameters.properties["shared_datasets"] = ",".join(shared_datasets)
+        if generate_features is not None:
+            if len(generate_features) > 1:
+                msg = bundle.get("too_many_generate_features")
+                self.logger.error(msg)
+                raise ValidationError(msg)
+            self.generate_features = generate_features
+            runtime_parameters = self.runtime_parameters or RuntimeParameters()
+            runtime_properties = runtime_parameters.properties or dict()
+            runtime_parameters["generate_features"] = ",".join(generate_features)
+            if round_embeddings is not None:
+                if not isinstance(round_embeddings, int) or round_embeddings < 0:
+                    msg = bundle.get("invalid_round_embeddings")
+                    self.logger.error(msg)
+                    raise ValidationError(msg)
+                self.round_embeddings = round_embeddings
+                runtime_properties["round_embeddings"] = round_embeddings
 
         self.passed_features: List[str] = []
         self.feature_names_ = []
@@ -1295,7 +1313,7 @@ class FeaturesEnricher(TransformerMixin):
             def enrich():
                 res, _ = self.__enrich(
                     df_with_original_index,
-                    validation_task.get_all_validation_raw_features(trace_id),
+                    validation_task.get_all_validation_raw_features(trace_id, metrics_calculation),
                     validated_X,
                     is_transform=True,
                 )

upgini/resource_bundle/strings.properties

@@ -56,6 +56,8 @@ no_connection_to_upgini=No connection to Upgini server https://search.upgini.com
 no_internet_connection=No internet connection from Jupyter server {} to initiate external data search with Upgini service, please try with Google Colab https://colab.research.google.com
 access_denied=Access denied
 unsupported_search_key=Search key {} not supported
+too_many_generate_features=Too many columns passed in `generate_features` argument. Only one column supported to generate features now
+invalid_round_embeddings=Argument `round_embeddings` should be non negative integer
 
 # Validation errors
     # params validation

{upgini-1.1.110a1.dist-info → upgini-1.1.111.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.110a1
+Version: 1.1.111
 Summary: Low-code feature search and enrichment library for machine learning
 Home-page: https://upgini.com/
 Author: Upgini Developers

{upgini-1.1.110a1.dist-info → upgini-1.1.111.dist-info}/RECORD

@@ -2,7 +2,7 @@ upgini/__init__.py,sha256=asENHgEVHQBIkV-e_0IhE_ZWqkCG6398U3ZLrNzAH6k,407
 upgini/ads.py,sha256=mre6xn44wcC_fg63iLT_kTh4mViZqR9AKRJZAtpQz8Y,2592
 upgini/dataset.py,sha256=J4KbHjvPKiAjmEs0GVeaHy-nZLAJXwpgYVeYKv-lGi0,41095
 upgini/errors.py,sha256=BqpvfhW2jJW5fa5KXj0alhXatGl-WK4xTl309-QNLp8,959
-upgini/features_enricher.py,sha256=HLsWY15UyzzaDA7xdfHYtOEmzeDOCyV0z_TBGC5ttQg,112299
+upgini/features_enricher.py,sha256=-1Oj8-Cy0LlCCNaHyPM2hIHPo7H9DPSL5cdKb5qT1QM,113360
 upgini/http.py,sha256=kgWj6wU1PbGPoGAbRvK35umXQ5zwEfEKeGy5Az0fss0,35479
 upgini/metadata.py,sha256=3PulMxpKFXAbmyqRh1GVzN5J_101YeZSH3h0LVe4ifc,5944
 upgini/metrics.py,sha256=uJhtGKgUUFnvdF16xscfe9AGDoDN6LqUV97RWDP39NU,14869
@@ -19,7 +19,7 @@ upgini/normalizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 upgini/normalizer/phone_normalizer.py,sha256=VIgLXuDuzzjPEXiy_LyDVLZKGaS7-le6Fh6T4D-TQDU,9930
 upgini/resource_bundle/__init__.py,sha256=M7GtS7KPQw9pinz8P2aQWXpSkD2YFwUPVGk1w92Pn84,7888
 upgini/resource_bundle/exceptions.py,sha256=KT-OnqA2J4OTfLjhbEl3KFZM2ci7EOPjqJuY_rXp3vs,622
-upgini/resource_bundle/strings.properties,sha256=66mJ8P3W3fBfhflIwSLrrJXvPvrE09CIqxDBqGuugi8,20042
+upgini/resource_bundle/strings.properties,sha256=If9Ej4T8ZsiqByTijcIAgQyAQci_y-089OOdD19k5DE,20261
 upgini/sampler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/sampler/base.py,sha256=X2PVsfZ3Rl7twpFDh5UWyxqY2K_jcMGxZ2NcHLwFRj4,6489
 upgini/sampler/random_under_sampler.py,sha256=whX_f_TtalHH8Seyn_7n3sX_TSiDHeYfALmme9saqDg,4082
@@ -39,8 +39,8 @@ upgini/utils/postal_code_utils.py,sha256=_8CR9tBqsPptQsmMUvnrCAmBaMIQSWH3JfJ4ly3
 upgini/utils/target_utils.py,sha256=3eHrDy_Dc9ozuOwHGnGA705m9glCxKmjB-DfLrflqiA,1370
 upgini/utils/track_info.py,sha256=O_oL4gy1jH0DVgtiUeZAW0YKCeRT4B_bzH_SZYkFaOE,4076
 upgini/utils/warning_counter.py,sha256=vnmdFo5-7GBkU2bK9h_uC0K0Y_wtfcYstxOdeRfacO0,228
-upgini-1.1.110a1.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.110a1.dist-info/METADATA,sha256=aS0uOMyQ5r6_KWwqAkd63eKasKR-Jri9IjF-6taOtGw,41110
-upgini-1.1.110a1.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-upgini-1.1.110a1.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
-upgini-1.1.110a1.dist-info/RECORD,,
+upgini-1.1.111.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.111.dist-info/METADATA,sha256=upni_OMilRlLQ0cxtue1OxoeMn3UDwNZCZ71Vi27FwM,41108
+upgini-1.1.111.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+upgini-1.1.111.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
+upgini-1.1.111.dist-info/RECORD,,