corrected string table

lewisfogden · Apr 29, 2024 · 6a30e3c · 6a30e3c
1 parent dd33077
commit 6a30e3c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ src-dir = "src"
 
 [project]
 name = "heavylight"
-version = "1.0.7"
+version = "1.0.8"
 authors = [
   { name="Lewis Fogden", email="[email protected]" },
   { name="Matthew Caseres", email="[email protected]"}

diff --git a/src/heavylight/__init__.py b/src/heavylight/__init__.py
@@ -4,4 +4,4 @@
 from .memory_optimized_cache import CacheGraph
 from .make_examples import make_example
 
-__version__ = '1.0.7'
+__version__ = '1.0.8'
diff --git a/src/heavylight/heavytables.py b/src/heavylight/heavytables.py
@@ -175,7 +175,7 @@ def __init__(self, df:pd.DataFrame, rectify: Union[bool, None]=False, safe: Unio
                 cols = df[col].unique()
                 string_mapper = StringLookup(cols)
                 self.mappers.append(string_mapper)
-                df_int_keys[col] = string_mapper.get(df_int_keys[col])
+                df_int_keys[col] = string_mapper.get(df_int_keys[col].values)
 
             elif col_type in ["str_unsafe", "band"]:
                 df_col = pd.DataFrame(df[col].unique(), columns=["band_name"]).reset_index().sort_values("band_name")