From 95ff43c18356e5aa54bf4ada092e919d02990e4f Mon Sep 17 00:00:00 2001
From: Konstantin Malanchev <hombit@gmail.com>
Date: Fri, 8 Nov 2024 15:58:03 -0500
Subject: [PATCH] Reimport *Search from root __init__.py

---
 docs/tutorials/join_catalogs.ipynb | 2 +-
 docs/tutorials/margins.ipynb       | 2 +-
 src/lsdb/__init__.py               | 1 +
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/docs/tutorials/join_catalogs.ipynb b/docs/tutorials/join_catalogs.ipynb
index 7167aa13..e9c2f35a 100644
--- a/docs/tutorials/join_catalogs.ipynb
+++ b/docs/tutorials/join_catalogs.ipynb
@@ -16,7 +16,7 @@
    "outputs": [],
    "source": [
     "import lsdb\n",
-    "from lsdb.core.search import ConeSearch"
+    "from lsdb import ConeSearch"
    ]
   },
   {
diff --git a/docs/tutorials/margins.ipynb b/docs/tutorials/margins.ipynb
index 784dae43..1c465c87 100644
--- a/docs/tutorials/margins.ipynb
+++ b/docs/tutorials/margins.ipynb
@@ -59,7 +59,7 @@
    },
    "outputs": [],
    "source": [
-    "from lsdb.core.search import BoxSearch\n",
+    "from lsdb import BoxSearch\n",
     "\n",
     "box = BoxSearch(ra=(179.5, 180.1), dec=(9.4, 10))\n",
     "\n",
diff --git a/src/lsdb/__init__.py b/src/lsdb/__init__.py
index ea7166e9..8ca04f22 100644
--- a/src/lsdb/__init__.py
+++ b/src/lsdb/__init__.py
@@ -1,4 +1,5 @@
 from ._version import __version__
 from .catalog import Catalog, MarginCatalog
+from .core.search import BoxSearch, ConeSearch, PolygonSearch
 from .loaders.dataframe.from_dataframe import from_dataframe
 from .loaders.hats.read_hats import read_hats