Ensured that lazy loading works with dataframe registration.

Author: ericmjl

pandas_flavor/__init__.py

@@ -1,17 +1,37 @@
 """Top-level API for pandas-flavor."""
-from .register import (
-    register_series_method,
-    register_series_accessor,
-    register_dataframe_method,
-    register_dataframe_accessor,
+import lazy_loader as lazy
+
+__getattr__, __dir__, __all__ = lazy.attach(
+    __name__,
+    submod_attrs={
+        "register": [
+            "register_series_method",
+            "register_series_accessor",
+            "register_dataframe_method",
+            "register_dataframe_accessor",
+        ],
+        "xarray": [
+            "register_xarray_dataarray_method",
+            "register_xarray_dataset_method",
+        ],
+    },
 )
-from .xarray import register_xarray_dataarray_method, register_xarray_dataset_method
+# from .register import (
+#     register_series_method,
+#     register_series_accessor,
+#     register_dataframe_method,
+#     register_dataframe_accessor,
+# )
+# from .xarray import (
+#     register_xarray_dataarray_method,
+#     register_xarray_dataset_method,
+# )
 
-__all__ = [
-    "register_series_method",
-    "register_series_accessor",
-    "register_dataframe_method",
-    "register_dataframe_accessor",
-    "register_xarray_dataarray_method",
-    "register_xarray_dataset_method",
-]
+# __all__ = [
+#     "register_series_method",
+#     "register_series_accessor",
+#     "register_dataframe_method",
+#     "register_dataframe_accessor",
+#     "register_xarray_dataarray_method",
+#     "register_xarray_dataset_method",
+# ]

pyproject.toml

@@ -1,5 +1,5 @@
 [tool.black]
-line-length = 88
+line-length = 79
 target-version = ['py39']
 include = '\.pyi?$'
 extend-exclude = '''