mypy final :o

cleanlab · May 1, 2024 · cb50cd6 · cb50cd6
1 parent daf0014
commit cb50cd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cleanlab_studio/utils/data_enrichment/enrich.py b/cleanlab_studio/utils/data_enrichment/enrich.py
@@ -137,7 +137,7 @@ def get_regex_matches(
     """
     regex_list = get_compiled_regex_list(regex)
     if isinstance(column_data, list):
-        return [get_regex_match(x, regex_list) for x in column_data]
+        return [get_regex_match(x, regex_list, disable_warnings) for x in column_data]
     elif isinstance(column_data, pd.Series):
         return column_data.apply(lambda x: get_regex_match(x, regex_list, disable_warnings))
     else: