fix: Spark source support table with pattern "db.table" (#2606)

kfiring · kevjumba · web-flow · commit 3ce51391e0b2 · 2022-07-26T12:05:15.000-07:00
* 1. fix: spark source support table with pattern "db.table" (#2605 #2605) Signed-off-by: Kevin Zhang <kzhang@tecton.ai> * Fix lint Signed-off-by: Kevin Zhang <kzhang@tecton.ai> Co-authored-by: leonpeng <kfiring> Co-authored-by: Kevin Zhang <kzhang@tecton.ai>
diff --git a/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark_source.py b/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark_source.py
@@ -177,7 +177,8 @@ def get_table_query_string(self) -> str:
         """Returns a string that can directly be used to reference this table in SQL"""
         if self.table:
             # Backticks make sure that spark sql knows this a table reference.
-            return f"`{self.table}`"
+            table = ".".join([f"`{x}`" for x in self.table.split(".")])
+            return table
         if self.query:
             return f"({self.query})"
 
