refactor: register pandas extension at top level

gosling/__init__.py

@@ -1,3 +1,18 @@
+import pandas as pd
+
 from gosling.schema import *
 from gosling.api import *
 from gosling.display import renderers
+
+
+@pd.api.extensions.register_dataframe_accessor("gos")  # type: ignore
+class GosAccessor:
+    def __init__(self, df: pd.DataFrame):
+        self._df = df
+
+    def csv(self, **kwargs):
+        from gosling.experimental.data import data_server
+
+        content = self._df.to_csv(index=False) or ""
+        url = data_server(content, extension="csv")
+        return dict(type="csv", url=url, **kwargs)

gosling/experimental/data.py

@@ -1,7 +1,6 @@
 import pathlib
 from typing import Callable, Dict, Optional, Union
 
-import pandas as pd
 import gosling.experimental._tilesets as tilesets
 from gosling.experimental._provider import Provider, Resource, TilesetResource
 from gosling.utils.core import _compute_data_hash
@@ -87,17 +86,6 @@ def load(url: Union[pathlib.Path, str], **kwargs):
     return load
 
 
-@pd.api.extensions.register_dataframe_accessor("gos")  # type: ignore
-class GosAccessor:
-    def __init__(self, df: pd.DataFrame):
-        self._df = df
-
-    def csv(self, **kwargs):
-        content = self._df.to_csv(index=False) or ""
-        url = data_server(content, extension="csv")
-        return dict(type="csv", url=url, **kwargs)
-
-
 # re-export json data util
 from gosling.data import json