Bump polars version to 1.0.0

kiesraad · Jul 2, 2024 · 3e55887 · 3e55887
1 parent d78a064
commit 3e55887
Show file tree

Hide file tree

Showing 3 changed files with 87 additions and 82 deletions.
diff --git a/Pipfile b/Pipfile
@@ -5,7 +5,7 @@ name = "pypi"
 
 [packages]
 defusedxml = "0.7.1"
-polars = "0.20.3"
+polars = "1.0.0"
 
 [dev-packages]
 pytest = "*"

diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/src/neighbourhood.py b/src/neighbourhood.py
@@ -224,10 +224,15 @@ def from_path(str_path: Optional[str]) -> Optional["NeighbourhoodData"]:
                 data = pl.scan_parquet(path)
             else:
                 return None
-        except Exception:
-            return None
 
-        if data.columns != ["zip_code", "neighbourhood_code", "ambiguous"]:
+            if data.collect_schema().names() != [
+                "zip_code",
+                "neighbourhood_code",
+                "ambiguous",
+            ]:
+                return None
+
+        except Exception:
             return None
 
         return NeighbourhoodData(data=data)