From 3b5d3e553cd193b2f098367e38dfbf7ff59d113d Mon Sep 17 00:00:00 2001
From: Raoul Schram <r.d.schram@uu.nl>
Date: Tue, 7 May 2024 11:55:25 +0200
Subject: [PATCH] Fix ruff tests

---
 metasyncontrib/disclosure/base.py   | 1 +
 metasyncontrib/disclosure/string.py | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/metasyncontrib/disclosure/base.py b/metasyncontrib/disclosure/base.py
index 9f85546..84fc762 100644
--- a/metasyncontrib/disclosure/base.py
+++ b/metasyncontrib/disclosure/base.py
@@ -1,5 +1,6 @@
 """Base class for all disclosure control distributions."""
 
+import polars as pl
 from metasyn.distribution.base import BaseDistribution
 
 
diff --git a/metasyncontrib/disclosure/string.py b/metasyncontrib/disclosure/string.py
index ff26a99..5309b34 100644
--- a/metasyncontrib/disclosure/string.py
+++ b/metasyncontrib/disclosure/string.py
@@ -3,11 +3,11 @@
 from metasyn.distribution.string import (
     FakerDistribution,
     FreeTextDistribution,
-    UniqueFakerDistribution,
     StringConstantDistribution,
+    UniqueFakerDistribution,
 )
 
-from metasyncontrib.disclosure.base import metadist_disclosure, DisclosureConstantMixin
+from metasyncontrib.disclosure.base import DisclosureConstantMixin, metadist_disclosure
 
 
 @metadist_disclosure()