post merge edits

Author: shobsi

bigframes/series.py

@@ -1193,7 +1193,8 @@ def rdivmod(self, other) -> Tuple[Series, Series]:  # type: ignore
         return (self.rfloordiv(other), self.rmod(other))
 
     def dot(self, other):
-        return (self * other).sum()
+        if isinstance(other, Series):
+            return (self * other).sum()
 
         # At this point other must be a DataFrame
         if len(other.columns.names) == 1:
@@ -1203,6 +1204,7 @@ def dot(self, other):
                 [(self * other[col]).sum() for col in other.columns],
                 index=other.columns,
                 name=self.name,
+                session=self._session,
             )
             result = mul_df.mask(na_df)
         else:

tests/system/small/test_series.py

@@ -3187,7 +3187,11 @@ def test_dot_df_with_na(scalars_dfs):
     )
 
     pd.testing.assert_series_equal(
-        bf_result.to_pandas(), pd_result, check_index_type=False, check_dtype=False
+        bf_result.to_pandas(),
+        pd_result,
+        check_index_type=False,
+        check_dtype=False,
+        check_exact=False,
     )