From 466f5e7332018dd62d74c154bc9d749bf39f89df Mon Sep 17 00:00:00 2001
From: gw265981 <184935895+gw265981@users.noreply.github.com>
Date: Mon, 3 Feb 2025 17:21:18 +0000
Subject: [PATCH 1/2] fix: add package API exports in top level `__all__`

---
 coreax/__init__.py | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/coreax/__init__.py b/coreax/__init__.py
index 50ba569dc..98e4aa4dc 100644
--- a/coreax/__init__.py
+++ b/coreax/__init__.py
@@ -24,7 +24,6 @@
 
 __version__ = "0.3.1"
 
-# pylint: disable=unused-import
 from coreax.approximation import (
     ANNchorApproximateKernel,
     ApproximateKernel,
@@ -43,4 +42,24 @@
 )
 from coreax.metrics import KSD, MMD
 from coreax.score_matching import KernelDensityMatching, SlicedScoreMatching
-# pylint: enable=unused-import
+
+__all__ = [
+    "ANNchorApproximateKernel",
+    "ApproximateKernel",
+    "MonteCarloApproximateKernel",
+    "NystromApproximateKernel",
+    "Coreset",
+    "Coresubset",
+    "Data",
+    "SupervisedData",
+    "LaplacianKernel",
+    "PCIMQKernel",
+    "ScalarValuedKernel",
+    "SquaredExponentialKernel",
+    "SteinKernel",
+    "UniCompositeKernel",
+    "KSD",
+    "MMD",
+    "KernelDensityMatching",
+    "SlicedScoreMatching",
+]

From 825b5f26186b3743c42c1e507a3041934c792b49 Mon Sep 17 00:00:00 2001
From: gw265981 <184935895+gw265981@users.noreply.github.com>
Date: Tue, 4 Feb 2025 13:56:25 +0000
Subject: [PATCH 2/2] fix: remove ruff ignore from `__init__.py`

---
 pyproject.toml | 1 -
 1 file changed, 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 2a7e188c3..d8b54e87b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -155,7 +155,6 @@ ignore = [
 ]
 
 [tool.ruff.lint.per-file-ignores]
-"coreax/__init__.py" = ["F401"]
 "documentation/source/conf.py" = ["E402"]
 "documentation/source/snippets/*" = ["D100", "F821"]
 "examples/*" = ["PLR0914", "PLR0915"]