Fix test

mroeschke · Jan 26, 2024 · d5e692a · d5e692a
1 parent 0f95a31
commit d5e692a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pandas/tests/io/test_sql.py b/pandas/tests/io/test_sql.py
@@ -99,14 +99,17 @@ def iris_table_metadata():
     import sqlalchemy
     from sqlalchemy import (
         Column,
-        Double,
-        Float,
         MetaData,
         String,
         Table,
     )
 
-    dtype = Double if Version(sqlalchemy.__version__) >= Version("2.0.0") else Float
+    if Version(sqlalchemy.__version__) >= Version("2.0.0"):
+        from sqlalchemy import Double as float_type
+    else:
+        from sqlalchemy import Float as float_type
+
+    dtype = float_type
     metadata = MetaData()
     iris = Table(
         "iris",