upgini 1.2.106a3956.dev1__py3-none-any.whl → 1.2.107__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.106a3956.dev1"
+__version__ = "1.2.107"

upgini/features_enricher.py

@@ -208,7 +208,7 @@ class FeaturesEnricher(TransformerMixin):
         self,
         search_keys: Optional[Dict[str, SearchKey]] = None,
         country_code: Optional[str] = None,
-        model_task_type: Optional[ModelTaskType] = None,
+        model_task_type: Optional[Union[ModelTaskType, str]] = None,
         api_key: Optional[str] = None,
         endpoint: Optional[str] = None,
         search_id: Optional[str] = None,
@@ -282,6 +282,8 @@ class FeaturesEnricher(TransformerMixin):
         self.country_code = country_code
         self.__validate_search_keys(search_keys, search_id)
 
+        if model_task_type is not None:
+            self.model_task_type = ModelTaskType.parse(model_task_type)
         self.model_task_type = model_task_type
         self.endpoint = endpoint
         self._search_task: Optional[SearchTask] = None

upgini/metadata.py

@@ -162,6 +162,15 @@ class ModelTaskType(Enum):
     def is_classification(self) -> bool:
         return self in [ModelTaskType.BINARY, ModelTaskType.MULTICLASS]
 
+    @staticmethod
+    def parse(task_type: Any) -> "ModelTaskType":
+        if isinstance(task_type, ModelTaskType):
+            return task_type
+        elif isinstance(task_type, str):
+            return ModelTaskType(task_type.upper())
+        else:
+            raise ValueError(f"Invalid task type: {task_type}")
+
 
 class ModelLabelType(Enum):
     GINI = "gini"

{upgini-1.2.106a3956.dev1.dist-info → upgini-1.2.107.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.106a3956.dev1
+Version: 1.2.107
 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.106a3956.dev1.dist-info → upgini-1.2.107.dist-info}/RECORD

@@ -1,11 +1,11 @@
-upgini/__about__.py,sha256=wMowjQ4NPzJq0VApX8igfNH42M4gS9xSnMJkWS4GHwk,34
+upgini/__about__.py,sha256=E6dOhnEHkanXoxanNLOMoP_tHzJfqbVNEEIpbqjineU,24
 upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=e6JDYTZ2AwC5aF-dqclKZKkiKrHo2f6cFmMQO2ZZmjM,32724
 upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
-upgini/features_enricher.py,sha256=dBCBqAhzVHqRB2b1sPy9wzSi9XtIzeb6uArmJhcjj_8,220370
+upgini/features_enricher.py,sha256=EVDTob2gmtQSpo1wYdExmd0sXDqeZtmg01zr5QHBXZc,220494
 upgini/http.py,sha256=DNcoS7qdxG0mOJn6I8r6O5I6XdIJTdzDzW3hkz3NgG4,45443
-upgini/metadata.py,sha256=vsbbHyPCP3Rs8WkeDgQg99uAA_zmsbDStAT-NwDYhO4,12455
+upgini/metadata.py,sha256=9_0lFEWPpIHRBW-xWYSEcwPzICTC6_bQ6dUUlE75Xns,12773
 upgini/metrics.py,sha256=V2SP6NS5bfFHzRqufeKVsCXME1yG4t_8Dmk2E3zKdYk,45715
 upgini/search_task.py,sha256=Q5HjBpLIB3OCxAD1zNv5yQ3ZNJx696WCK_-H35_y7Rs,17912
 upgini/spinner.py,sha256=4iMd-eIe_BnkqFEMIliULTbj6rNI2HkN_VJ4qYe0cUc,1118
@@ -71,7 +71,7 @@ upgini/utils/target_utils.py,sha256=i3Xt5l9ybB2_nF_ma5cfPuL3OeFTs2dY2xDI0p4Azpg,
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
 upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
-upgini-1.2.106a3956.dev1.dist-info/METADATA,sha256=8gm0u1avCVEV1kBGkO9qYEFZHuRJ-lv_c4i0CfyO1Hg,49539
-upgini-1.2.106a3956.dev1.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-upgini-1.2.106a3956.dev1.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.106a3956.dev1.dist-info/RECORD,,
+upgini-1.2.107.dist-info/METADATA,sha256=P8FTzwQgOC9b_KOsV32-zHtNmsdXDnT3xyAO2Ggin74,49529
+upgini-1.2.107.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.107.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.107.dist-info/RECORD,,

{upgini-1.2.106a3956.dev1.dist-info → upgini-1.2.107.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.24.2
+Generator: hatchling 1.25.0
 Root-Is-Purelib: true
 Tag: py3-none-any