upgini 1.1.253a3__py3-none-any.whl → 1.1.253a5__py3-none-any.whl

upgini/features_enricher.py

@@ -3022,6 +3022,7 @@ class FeaturesEnricher(TransformerMixin):
         return result_train, result_eval_sets
 
     def __prepare_feature_importances(self, trace_id: str, x_columns: List[str], silent=False):
+        llm_source = "LLM with external data augmentation"
         if self._search_task is None:
             raise NotFittedError(self.bundle.get("transform_unfitted_enricher"))
         features_meta = self._search_task.get_all_features_metadata_v2()
@@ -3076,6 +3077,8 @@ class FeaturesEnricher(TransformerMixin):
                     return ""
                 elif not link:
                     return value
+                elif value == llm_source:
+                    return value
                 else:
                     return f"<a href='{link}' target='_blank' rel='noopener noreferrer'>{value}</a>"
 
@@ -3092,7 +3095,7 @@ class FeaturesEnricher(TransformerMixin):
                 provider = to_anchor("https://upgini.com", "Upgini")
 
             internal_source = feature_meta.data_source or (
-                "LLM with external data augmentation"
+                llm_source
                 if not feature_meta.name.endswith("_country") and not feature_meta.name.endswith("_postal_code")
                 else ""
             )

upgini/utils/target_utils.py

@@ -152,7 +152,7 @@ def balance_undersample(
         # fill up to min_sample_threshold by majority class
         minority_class = df[df[target_column] == min_class_value]
         majority_class = df[df[target_column] != min_class_value]
-        sample_size = min(len(majority_class, min_sample_threshold - min_class_count))
+        sample_size = min(len(majority_class), min_sample_threshold - min_class_count)
         sampled_majority_class = majority_class.sample(
             n=sample_size, random_state=random_state
         )

{upgini-1.1.253a3.dist-info → upgini-1.1.253a5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.253a3
+Version: 1.1.253a5
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers

{upgini-1.1.253a3.dist-info → upgini-1.1.253a5.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=sL1w8dVmXkuV5PWaBTUpz8X8joW6mIOLyPbSAlSbpR0,45576
 upgini/errors.py,sha256=pdzQl3MKuK52yvncxMWMRWeSIOGhUFzpQoszoRFBOk0,958
-upgini/features_enricher.py,sha256=wsMb_LljaJPxa-UqokN80CGzOfXSFhXt1NbZfZHw9YU,171715
+upgini/features_enricher.py,sha256=1f92VFY3ZyVGdTTY81BPQs6OjrSAQJc0U41C6QsWLQ0,171822
 upgini/fingerprint.js,sha256=VygVIQlN1v4NGZfjHqtRogOw8zjTnnMNJg_f7M5iGQU,33442
 upgini/http.py,sha256=eSG4gOpmCGlXmB6KIPNzAG8tRZNUjyYpMeUeHw_2li4,42264
 upgini/metadata.py,sha256=fwVxtkR6Mn4iRoOqV6BfMJvJrx65I3YwZUMbZjhPyOI,9673
@@ -52,11 +52,11 @@ upgini/utils/phone_utils.py,sha256=JNSkF8G6mgsN8Czy11pamaJdsY6rBINEMpi7jbVt_RA,4
 upgini/utils/postal_code_utils.py,sha256=_8CR9tBqsPptQsmMUvnrCAmBaMIQSWH3JfJ4ly3x_zs,409
 upgini/utils/progress_bar.py,sha256=iNXyqT3vKCeHpfiG5HHwr7Lk2cTtKViM93Fl8iZnjGc,1564
 upgini/utils/sklearn_ext.py,sha256=fvuTWJ5AnT3ED9KSaQu_yIgW2JR19hFlaGDoVP3k60g,44027
-upgini/utils/target_utils.py,sha256=OzW1dlhW0tQj5FBR-iIIjdpzqIGfGFRoYePppP8yRhw,7204
+upgini/utils/target_utils.py,sha256=WVhhxpQVvnhsDV7ctlds51VFg7hz59S_MFUSoRZFszw,7204
 upgini/utils/track_info.py,sha256=EPcJ13Jqa17_T0JjM37Ac9kWDz5Zk0GVsIZKutOb8aU,5207
 upgini/utils/warning_counter.py,sha256=dIWBB4dI5XRRJZudvIlqlIYKEiwLLPcXarsZuYRt338,227
-upgini-1.1.253a3.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.253a3.dist-info/METADATA,sha256=mFmkLSCSitZB6j7-e8UKEdt7woYNlzGaKIeMGCnlpIo,48158
-upgini-1.1.253a3.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-upgini-1.1.253a3.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
-upgini-1.1.253a3.dist-info/RECORD,,
+upgini-1.1.253a5.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.253a5.dist-info/METADATA,sha256=sYd-VQbiUxqSOb9cgjavIjagvPxM1uyO7fgAT97UZg8,48158
+upgini-1.1.253a5.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+upgini-1.1.253a5.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
+upgini-1.1.253a5.dist-info/RECORD,,