fix mypy errors for newer pandas

Author: erindru

sqlmesh/core/test/definition.py

@@ -648,7 +648,9 @@ def _create_df(
             if partial:
                 columns = referenced_columns
 
-        return pd.DataFrame.from_records(rows, columns=columns)
+        return pd.DataFrame.from_records(
+            rows, columns=[str(c) for c in columns] if columns else None
+        )
 
     def _add_missing_columns(
         self, query: exp.Query, all_columns: t.Optional[t.Collection[str]] = None

tests/integrations/github/cicd/test_integration.py

@@ -37,11 +37,9 @@ def get_environment_objects(controller: GithubController, environment: str) -> t
 
 
 def get_num_days_loaded(controller: GithubController, environment: str, model: str) -> int:
-    return int(
-        controller._context.engine_adapter.fetchdf(
-            f"SELECT distinct event_date FROM sushi__{environment}.{model}"
-        ).count()
-    )
+    return controller._context.engine_adapter.fetchdf(
+        f"SELECT distinct event_date FROM sushi__{environment}.{model}"
+    ).shape[0]
 
 
 def get_columns(