Hotfix search keys check

c3p0-upgini · c3p0-upgini · commit 6c4295663ce0 · 2022-08-22T12:00:59.000+03:00
diff --git a/setup.py b/setup.py
@@ -39,7 +39,7 @@ def send_log(msg: str):
     send_log("Start setup PyLib")
     setup(
         name="upgini",
-        version="1.1.17",
+        version="1.1.18",
         description="Low-code feature search and enrichment library for machine learning",
         long_description=(here / "README.md").read_text(encoding="utf-8"),
         long_description_content_type="text/markdown",
diff --git a/src/upgini/features_enricher.py b/src/upgini/features_enricher.py
@@ -1031,7 +1031,11 @@ def __prepare_search_keys(self, x: pd.DataFrame):
         self.search_keys = valid_search_keys
 
         using_keys = self.__using_search_keys()
-        if len(using_keys.values()) == 1 and next(iter(using_keys.values())) == SearchKey.DATE:
+        if (
+            len(using_keys.values()) == 1
+            and self.country_code is None
+            and next(iter(using_keys.values())) == SearchKey.DATE
+        ):
             msg = (
                 "WARNING: You have started the search with the Date key only. "
                 "Try to add the Country and/or Postal Code keys to your dataset so that the search engine gets access "
