From 0cd5614321dfdab486819e49e633b2b8be1c1350 Mon Sep 17 00:00:00 2001
From: taylorfturner <taylorfturner@gmail.com>
Date: Wed, 12 Jun 2024 13:58:40 -0400
Subject: [PATCH] update

---
 dataprofiler/__init__.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/dataprofiler/__init__.py b/dataprofiler/__init__.py
index 5f218bd85..2e89d3e2b 100644
--- a/dataprofiler/__init__.py
+++ b/dataprofiler/__init__.py
@@ -20,6 +20,22 @@
 from .validators.base_validators import Validator
 from .version import __version__
 
+try:
+    import snappy
+except ImportError:
+    import warnings
+
+    warnings.warn(
+        "Snappy must be installed to use parquet/avro datasets."
+        "\n\n"
+        "For macOS use Homebrew:\n"
+        "\t`brew install snappy`"
+        "\n\n"
+        "For linux use apt-get:\n`"
+        "\tsudo apt-get -y install libsnappy-dev`\n",
+        ImportWarning,
+    )
+
 
 def set_seed(seed=None):
     # also check it's an integer