Importing demo submodule by default

Author: amontanez24

sdv/datasets/__init__.py

@@ -1 +1,5 @@
 """Dataset loading and managing module."""
+
+from sdv.datasets import demo, local
+
+__all__ = ['demo', 'local']

tests/unit/datasets/test_demo.py

@@ -1144,7 +1144,7 @@ def test_download_demo_skips_non_csv_in_memory_no_warning(mock_list, mock_get):
         data, _ = download_demo('single_table', 'mix')
 
     assert any(warn_msg in str(warn_record) for warn_record in rec)
-        
+
     expected = pd.DataFrame({'id': [1, 2], 'name': ['a', 'b']})
     pd.testing.assert_frame_equal(data, expected)