fix: Make sure schema is used when calling get_table_query_string method for Snowflake datasource (feast-dev#4131)

TomSteenbergen · web-flow · commit c1579c77324c · 2024-04-25T07:20:19.000-05:00
* Fix get_table_query_string method for Snowflake datasource

Signed-off-by: TomSteenbergen &lt;tomsteenbergen1995@gmail.com&gt;

* Add quotes to table string

Signed-off-by: TomSteenbergen &lt;tomsteenbergen1995@gmail.com&gt;

---------

Signed-off-by: TomSteenbergen &lt;tomsteenbergen1995@gmail.com&gt;
diff --git a/sdk/python/feast/infra/offline_stores/snowflake_source.py b/sdk/python/feast/infra/offline_stores/snowflake_source.py
@@ -191,8 +191,10 @@ def validate(self, config: RepoConfig):
 
     def get_table_query_string(self) -> str:
         """Returns a string that can directly be used to reference this table in SQL."""
-        if self.database and self.table:
+        if self.database and self.schema and self.table:
             return f'"{self.database}"."{self.schema}"."{self.table}"'
+        elif self.schema and self.table:
+            return f'"{self.schema}"."{self.table}"'
         elif self.table:
             return f'"{self.table}"'
         else:
