diff --git a/intake_parquet/__init__.py b/intake_parquet/__init__.py index b607ed8..489cdde 100644 --- a/intake_parquet/__init__.py +++ b/intake_parquet/__init__.py @@ -3,4 +3,5 @@ __version__ = get_versions()['version'] del get_versions +import intake from intake_parquet.source import ParquetSource diff --git a/tests/test_discovery.py b/tests/test_discovery.py index 0cb0bd6..62bd6da 100644 --- a/tests/test_discovery.py +++ b/tests/test_discovery.py @@ -8,4 +8,5 @@ def test_discovery(): # For awhile we expect a PendingDeprecationWarning due to # do_pacakge_scan=True. But we should *not* get a FutureWarning. for record in record.list: - assert not isinstance(record.message, FutureWarning) + if isinstance(record.message, FutureWarning): + assert "parquet" not in str(record.message)