fix sf df naming

capitalone · Feb 18, 2025 · 5b780c4 · 5b780c4
1 parent 96ad882
commit 5b780c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datacompy/snowflake.py b/datacompy/snowflake.py
@@ -163,7 +163,7 @@ def df1(self, df1: tuple[Union[str, "sp.DataFrame"], str | None]) -> None:
             if len(table_name) != 3:
                 errmsg = f"{df} is not a valid table name. Be sure to include the target db and schema."
                 raise ValueError(errmsg)
-            self.df1_name = df_name.upper() if df_name else table_name[2]
+            self.df1_name = df_name.upper() if df_name else "__".join(table_name)
             self._df1 = self.session.table(df)
         else:
             self._df1 = df
@@ -184,7 +184,7 @@ def df2(self, df2: tuple[Union[str, "sp.DataFrame"], str | None]) -> None:
             if len(table_name) != 3:
                 errmsg = f"{df} is not a valid table name. Be sure to include the target db and schema."
                 raise ValueError(errmsg)
-            self.df2_name = df_name.upper() if df_name else table_name[2]
+            self.df2_name = df_name.upper() if df_name else "__".join(table_name)
             self._df2 = self.session.table(df)
         else:
             self._df2 = df