[Fix] add_data() function failed in keplergl-jupyter (#2901)

Author: lixun910

bindings/kepler.gl-jupyter/js/package.json

@@ -1,6 +1,6 @@
 {
   "name": "keplergl-jupyter",
-  "version": "0.3.5",
+  "version": "0.3.6",
   "description": "This is a simple jupyter widget for kepler.gl, an advanced geo-spatial visualization tool, to render large-scale interactive maps.",
   "author": "Shan He",
   "license": "MIT",

bindings/kepler.gl-jupyter/keplergl/_version.py

@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: MIT
 # Copyright contributors to the kepler.gl project
 
-version_info = (0, 3, 5, 'final', 0)
+version_info = (0, 3, 6, 'final', 0)
 
 _specifier_ = {'alpha': 'a', 'beta': 'b', 'candidate': 'rc', 'final': ''}
 
@@ -17,4 +17,4 @@
 #
 # The major version needs to match that of the JS package.
 # Note: this follows the semver format, which is used to match the JS package version in keplergl-plugin.js
-EXTENSION_SPEC_VERSION = '0.3.5'
+EXTENSION_SPEC_VERSION = '0.3.6'

bindings/kepler.gl-jupyter/keplergl/keplergl.py

@@ -209,26 +209,15 @@ def add_data(self, data, name="unnamed", use_arrow=False):
         ''' Send data to Voyager
 
         Inputs:
-        - data string, can be a csv string or json string
+        - data string, can be a dataframe, csv string or json string
         - name string
 
         Example of use:
             keplergl.add_data(data_string, name="data_1")
         '''
+        normalized = _normalize_data(data, use_arrow)
         copy = self.data.copy()
-
-        # assume data is a GeoJSON or CSV string, convert it to arrow if use_arrow is True
-        if use_arrow:
-            global g_use_arrow
-            g_use_arrow = use_arrow
-        try:
-            gdf = geopandas.read_file(data, driver='GeoJSON')
-            copy.update({name: gdf})
-        except Exception:
-            # if it fails, assume it is a csv string
-            # load csv string to a dataframe
-            df = pd.read_csv(data)
-            copy.update({name: df})
+        copy.update({name: normalized})
 
         self.data = copy