upgini 1.2.6__tar.gz → 1.2.7__tar.gz

{upgini-1.2.6 → upgini-1.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.6
+Version: 1.2.7
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/

upgini-1.2.7/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.7"

{upgini-1.2.6 → upgini-1.2.7}/src/upgini/data_source/data_source_publisher.py

@@ -64,6 +64,7 @@ class DataSourcePublisher:
         date_features: Optional[List[str]] = None,
         date_vector_features: Optional[List[str]] = None,
         generate_runtime_embeddings: Optional[List[str]] = None,
+        exclude_raw: Optional[List[str]] = None,
         _force_generation=False,
         _silent=False,
     ) -> str:
@@ -88,6 +89,8 @@ class DataSourcePublisher:
         features_for_embeddings - optional list of str - list of features that should be used for GPT features
             generation
 
+        exclude_raw - optional list of str - list of features that should NOT be used as raw features
+
         ...
 
         data_table_id_to_replace - optional str - id of registered ADS that should be replaced by new table
@@ -166,6 +169,8 @@ class DataSourcePublisher:
                     request["dateVectorFeatures"] = date_vector_features
                 if generate_runtime_embeddings is not None:
                     request["generateRuntimeEmbeddingsFeatures"] = generate_runtime_embeddings
+                if exclude_raw is not None:
+                    request["excludeRaw"] = exclude_raw
                 self.logger.info(f"Start registering data table {request}")
 
                 task_id = self._rest_client.register_ads(request, trace_id)
@@ -281,6 +286,7 @@ class DataSourcePublisher:
         date_vector_features: Optional[List[str]] = None,
         exclude_from_autofe_generation: Optional[List[str]] = None,
         generate_runtime_embeddings: Optional[List[str]] = None,
+        exclude_raw: Optional[List[str]] = None,
     ):
         trace_id = str(uuid.uuid4())
         with MDC(trace_id=trace_id):
@@ -336,6 +342,8 @@ class DataSourcePublisher:
                     request["excludeFromGenerationFeatures"] = exclude_from_autofe_generation
                 if generate_runtime_embeddings is not None:
                     request["generateRuntimeEmbeddingsFeatures"] = generate_runtime_embeddings
+                if exclude_raw is not None:
+                    request["excludeRaw"] = exclude_raw
                 self.logger.info(f"Activating data tables with request {request}")
 
                 self._rest_client.activate_datatables(request, trace_id)

upgini-1.2.6/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.6"