Dtypes is deprecated/removed in polars 0.20.31 (#41)

* Dtypes is deprecated/removed in polars 0.20.31
* Fix NA distribution

.github/workflows/python-package.yml

@@ -19,7 +19,7 @@ jobs:
         metasyn-version: [ "git+https://github.com/sodascience/metasyn.git@main"]
         include:
           - python-version: "3.11"
-            metasyn-version: "metasyn==1.0.0"
+            metasyn-version: "metasyn==1.0.1"
 
     steps:
     - uses: actions/checkout@v4

metasyncontrib/disclosure/base.py

@@ -25,7 +25,7 @@ class DisclosureConstantMixin(BaseDistribution):
     """Mixin class to overload fit method for constant distributions."""
 
     @classmethod
-    def fit(cls, series, *args, partition_size: int = 11, **kwargs) -> BaseDistribution:
+    def fit(cls, series, *args, partition_size: int = 11, **kwargs) -> BaseDistribution:  #pylint: disable=unused-argument
         """Fit constant distributions with disclosure control rules in place."""
         # NB: dominance rule ensures that constant distribution is essentially never
         # allowed under formal disclosure control. Always return default distribution.

metasyncontrib/disclosure/na.py

@@ -9,3 +9,7 @@
 @metadist_disclosure()
 class DisclosureNA(NADistribution):
     """Disclosure version of NA distribution."""
+
+    @classmethod
+    def _fit(cls, values, partition_size: int = 11):
+        return cls.default_distribution()

metasyncontrib/disclosure/utils.py

@@ -6,7 +6,11 @@
 
 import numpy as np
 import polars as pl
-from numpy.core._exceptions import UFuncTypeError
+
+try:
+    from numpy.core._exceptions import UFuncTypeError  # type: ignore
+except ImportError:
+    from numpy._core._exceptions import UFuncTypeError  # type: ignore
 
 
 def _compute_dominance(block_values, reverse=False):

pyproject.toml

@@ -24,9 +24,9 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
 ]
 dependencies = [
-	"metasyn>=0.8.0",
+	"metasyn>=1.0.1",
 	"polars",
-	"numpy>=1.20",
+	"numpy>=1.20,<2.0",
 ]
 dynamic = ["version"]